llm-demo1-finalfinal

Runtime error

storresbusquets commited on Sep 17, 2023

Commit

b0daf63

1 Parent(s): a57c372

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ access_token = os.getenv("Llama2")
 def greet(token, text):
     model = "meta-llama/Llama-2-7b-chat-hf"
-    tokenizer = AutoTokenizer.from_pretrained(model)
     pipeline = transformers.pipeline(
     "text-generation",

 def greet(token, text):
     model = "meta-llama/Llama-2-7b-chat-hf"
+    tokenizer = AutoTokenizer.from_pretrained(model, token=access_token)
     pipeline = transformers.pipeline(
     "text-generation",