Fix

app.py

@@ -6,7 +6,11 @@ from PIL import Image
 import gradio as gr
 import spaces
 import torch
-from transformers import AutoProcessor, Gemma3ForConditionalGeneration, TextIteratorStreamer
+from transformers import (
+    AutoProcessor,
+    Gemma3ForConditionalGeneration,
+    TextIteratorStreamer,
+)
 import subprocess
 
 subprocess.run(
@@ -107,9 +111,13 @@ def generate(
 
     # Use the processor to create the prompt and preprocess images
     prompt = processor.apply_chat_template(conversation, add_generation_prompt=True)
-    inputs = processor(text=prompt, images=all_images, return_tensors="pt").to(
-        model.device
-    )
+
+    if len(all_images) > 0:
+        inputs = processor(text=prompt, images=all_images, return_tensors="pt").to(
+            model.device
+        )
+    else:
+        inputs = processor(text=prompt, return_tensors="pt").to(model.device)
 
     # Set up the streamer for text generation
     streamer = TextIteratorStreamer(