Spaces:
Running
Running
luanpoppe
commited on
Commit
·
d8142ab
1
Parent(s):
fc85f25
novo commit
Browse files- endpoint_teste/views.py +5 -4
endpoint_teste/views.py
CHANGED
@@ -45,10 +45,11 @@ def getTeste(request):
|
|
45 |
# print('output: ', output)
|
46 |
# print('output: ', dir(output))
|
47 |
llm = HuggingFaceEndpoint(
|
48 |
-
|
49 |
-
|
50 |
-
|
51 |
-
|
|
|
52 |
)
|
53 |
result = llm.invoke("Hugging Face is")
|
54 |
print('result: ', result)
|
|
|
45 |
# print('output: ', output)
|
46 |
# print('output: ', dir(output))
|
47 |
llm = HuggingFaceEndpoint(
|
48 |
+
repo_id="meta-llama/Meta-Llama-3-8B-Instruct",
|
49 |
+
task="text-generation",
|
50 |
+
max_new_tokens=100,
|
51 |
+
do_sample=False,
|
52 |
+
huggingfacehub_api_token = os.environ.get("HUGGINGFACEHUB_API_TOKEN")
|
53 |
)
|
54 |
result = llm.invoke("Hugging Face is")
|
55 |
print('result: ', result)
|