Hugging Face's logo

Spaces:
Kush1
/
LLAMA-2
Runtime error

App Files Community
Kush1 commited on
Commit
bf098de
·
1 Parent(s): c2b376f

Changes in app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ def get_llama_response(pipeline,prompt):
36
  print(sequences[0]['generated_text'])
37
 
38
 
39
- pipeline = AutoTokenizer.from_pretrained(model)
40
  #prompt="Can you help me to write rest api endpoints in python ?"
41
  response = get_llama_response(pipeline,prompt)
42
 
 
36
  print(sequences[0]['generated_text'])
37
 
38
 
39
+ pipeline = AutoTokenizer.from_pretrained(model, token=HF_TOKEN)
40
  #prompt="Can you help me to write rest api endpoints in python ?"
41
  response = get_llama_response(pipeline,prompt)
42