UPdating app.py

app.py

@@ -7,7 +7,7 @@ import gradio as gr
 torch.set_default_device("cuda")
 
 
-model = "deepapaikar/llama_mistral"
+model = "deepapaikar/katzbot-phi2"
 
 pipeline = transformers.pipeline(
     "text-generation",
@@ -40,11 +40,11 @@ def gradio_predict(question, token):
 # Define the Gradio interface
 iface = gr.Interface(
     fn=gradio_predict,
-    inputs=[gr.Textbox(label="Question", placeholder="e.g. What are the colors of the bus in the image?", scale=4),
+    inputs=[gr.Textbox(label="Question", placeholder="e.g. Where is Yeshiva University located?", scale=4),
            gr.Slider(2, 100, value=25, label="Token Count", info="Choose between 2 and 100")],
     outputs=gr.TextArea(label="Answer"),
     title="KatzBot",
-    description="Llama - Mistral Merge",
+    description="Phi2-trial1",
 )
 
 # Launch the app