Spaces:

vsrinivas
/

ChatBot_by_SrinivasV

Sleeping

vsrinivas commited on Sep 22, 2023

Commit

77fa3ae

1 Parent(s): 78e8b35

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ checkpoint = "tiiuae/falcon-40b-instruct"
 model = AutoModelForCausalLM.from_pretrained(
     checkpoint, device_map="auto", offload_folder="offload", trust_remote_code=True,)
-tokenizer = AutoTokenizer.from_pretrained(checkpoint, trust_remote_code=True,")
 pipeline = transformers.pipeline(
     "text-generation",

 model = AutoModelForCausalLM.from_pretrained(
     checkpoint, device_map="auto", offload_folder="offload", trust_remote_code=True,)
+tokenizer = AutoTokenizer.from_pretrained(checkpoint, trust_remote_code=True,)
 pipeline = transformers.pipeline(
     "text-generation",