luanpoppe commited on
Commit
d8142ab
·
1 Parent(s): fc85f25

novo commit

Browse files
Files changed (1) hide show
  1. 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
- repo_id="meta-llama/Meta-Llama-3-8B-Instruct",
49
- task="text-generation",
50
- max_new_tokens=100,
51
- do_sample=False,
 
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)