Update app.py

app.py

@@ -49,13 +49,19 @@ with gr.Blocks() as demo:
             prompt_input_custom = gr.Textbox(label="Enter your custom prompt")
         generate_button_custom = gr.Button("Generate")
         output_text_custom = gr.Textbox(label="Response")
-        generate_button_custom.click(fn=generate, inputs=[image_input_custom, prompt_input_custom], outputs=output_text_custom)
+
+        def custom_generate(image, prompt):
+            return generate(image, prompt)
+
+        generate_button_custom.click(fn=custom_generate, inputs=[image_input_custom, prompt_input_custom], outputs=output_text_custom)
+
         example_prompt = "65 y/m Chronic cough and weight loss x 6 months. Chest X-rays normal. Consulted multiple pulmonologists with not much benefit. One wise pulmonologist thinks of GERD and sends him to the Gastro department. Can you name the classical finding here?"
         example_image_path = os.path.join(os.path.dirname(__file__), "hegde.jpg")
         gr.Examples(
             examples=[[example_image_path, example_prompt]],
             inputs=[image_input_custom, prompt_input_custom],
             outputs=[output_text_custom],
+            fn=custom_generate,
             cache_examples=True
         )