Update src/app/model.py

src/app/model.py

@@ -3,6 +3,7 @@ import sys
 import subprocess
 from typing import Optional
 from PIL import Image
+import gradio as gr
 import spaces
 from transformers import AutoProcessor, AutoModelForCausalLM
 
@@ -51,6 +52,10 @@ def run_example(
         str: The parsed answer generated by the model.
     """
     try:
+        # Check if image is None
+        if image is None:
+            raise gr.Error("Please provide an image.")
+
         # If there is no text input, use the task prompt as the prompt
         prompt = task_prompt if text_input is None else task_prompt + text_input