Update app.py

app.py

@@ -25,9 +25,9 @@ model_name = "microsoft/phi-2"
 phi2_text = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 
-def example_inference(input_text, count, image, img_qn, audio):
+def example_inference(input_text, count) #, image, img_qn, audio):
     pred_text = textMode(input_text, count)
-    return pred_text, "in progress", "in progress"
+    return pred_text #, "in progress", "in progress"
 
 
 
@@ -83,10 +83,10 @@ with gr.Blocks() as demo:
 
     gr.Examples(
         examples=[
-            ["What is a large language model?","50","","Describe the image",""]            
+            ["What is a large language model?","50"] #,"","Describe the image",""]            
         ],
-        inputs=[text_input, text_input_count, image_input, image_text_input, audio_input],
-        outputs=[text_output, image_text_output, audio_text_output],
+        inputs=[text_input, text_input_count] #, image_input, image_text_input, audio_input],
+        outputs=[text_output] #, image_text_output, audio_text_output],
         fn=example_inference,
     )