luanpoppe commited on
Commit
aeb38f0
·
1 Parent(s): deb05dc

feat: tentando utilizar um model de llama3 do hugging face

Browse files
Files changed (1) hide show
  1. endpoint_teste/views.py +13 -2
endpoint_teste/views.py CHANGED
@@ -1,4 +1,4 @@
1
- import tempfile, os
2
  from rest_framework import viewsets, filters
3
  from django_filters.rest_framework import DjangoFilterBackend
4
  from endpoint_teste.models import EndpointTesteModel
@@ -31,7 +31,18 @@ def getTeste(request):
31
  "Resposta": resposta_llm
32
  })
33
  if request.method == "GET":
34
- return Response("Fazendo um GET")
 
 
 
 
 
 
 
 
 
 
 
35
 
36
  @api_view(["POST"])
37
  def getPDF(request):
 
1
+ import tempfile, os, requests
2
  from rest_framework import viewsets, filters
3
  from django_filters.rest_framework import DjangoFilterBackend
4
  from endpoint_teste.models import EndpointTesteModel
 
31
  "Resposta": resposta_llm
32
  })
33
  if request.method == "GET":
34
+ API_URL = "https://api-inference.huggingface.co/models/meta-llama/Meta-Llama-3-8B"
35
+ headers = {"Authorization": "Bearer hf_xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}
36
+ def query(payload):
37
+ response = requests.post(API_URL, headers=headers, json=payload)
38
+ return response.json()
39
+
40
+ output = query({
41
+ "inputs": "Can you please let us know more details about your something I don't know",
42
+ })
43
+ print('output: ', output)
44
+ print('output: ', dir(output))
45
+ return Response(output)
46
 
47
  @api_view(["POST"])
48
  def getPDF(request):