Spaces:
Running
Running
luanpoppe
commited on
Commit
·
aeb38f0
1
Parent(s):
deb05dc
feat: tentando utilizar um model de llama3 do hugging face
Browse files- 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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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):
|