Update app.py

app.py

@@ -5,10 +5,10 @@ import gradio as gr
 
 # checkpoint  = "tiiuae/falcon-40b-instruct"
 # checkpoint ="tiiuae/falcon-7b-instruct"
-checkpoint =  "tiiuae/falcon-7b"
+# checkpoint =  "tiiuae/falcon-7b"
 # checkpoint = "mistralai/Mixtral-8x7B-Instruct-v0.1"
 # checkpoint = "gpt2"
-# checkpoint = "vsrinivas/FalconLite2"
+checkpoint = "vsrinivas/FalconLite2"
 
 model = AutoModelForCausalLM.from_pretrained(
     checkpoint, device_map="auto",