Update app.py

app.py

@@ -3,14 +3,14 @@ import transformers
 import torch
 import gradio as gr
 
-# checkpoint = "tiiuae/falcon-40b-instruct"
+checkpoint = "tiiuae/falcon-40b-instruct"
 # checkpoint = "tiiuae/falcon-7b-instruct"
-checkpoint = "microsoft/phi-1_5"
 
 model = AutoModelForCausalLM.from_pretrained(
-    checkpoint, device_map="auto", offload_folder="offload", trust_remote_code=True)
+    checkpoint, device_map="auto", offload_folder="offload", trust_remote_code=True,)
+
+tokenizer = AutoTokenizer.from_pretrained(checkpoint, trust_remote_code=True,")
 
-tokenizer = AutoTokenizer.from_pretrained(checkpoint, )
 pipeline = transformers.pipeline(
     "text-generation",
     model=model,