Spaces:

dzekoh
/

ChatBot

Runtime error

dzekoh commited on Aug 1, 2023

Commit

d1076ba

1 Parent(s): 106d763

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,8 +9,8 @@ examples = [["How are you?"]]
 tokenizer = AutoTokenizer.from_pretrained("tiiuae/falcon-40b-instruct")
-model = AutoModelForCausalLM.from_pretrained("tiiuae/falcon-40b-instruct")
-trust_remote_code=True
 def predict(input, history=[]):
     # tokenize the new input sentence

 tokenizer = AutoTokenizer.from_pretrained("tiiuae/falcon-40b-instruct")
+model = AutoModelForCausalLM.from_pretrained("tiiuae/falcon-40b-instruct", trust_remote_code=True)
 def predict(input, history=[]):
     # tokenize the new input sentence