Update src/model.py

src/model.py

@@ -1,4 +1,5 @@
 # Importing necessary libraries
+import gradio as gr
 import subprocess
 import spaces
 from transformers import AutoProcessor, AutoModelForCausalLM
@@ -28,6 +29,10 @@ def run_example(task_prompt, image, text_input=None):
     Returns:
         str: The parsed answer generated by the model.
     """
+    # Check if the image is provided
+    if not image:
+        gr.error("No image provided")
+        
     # If there is no text input, use the task prompt as the prompt
     if text_input is None:
         prompt = task_prompt