Update app.py

app.py

@@ -109,7 +109,7 @@ def run_prediction():
         generation = model.generate(**batch, max_new_tokens=100)
     decoded = processor.batch_decode(generation, skip_special_tokens=True)[0]
 
-    image = raw_image[0]
+    image = raw_image  # ✅ FIXED: raw_image is already a PIL.Image
     width, height = image.size
 
     result_image = visualize_bounding_boxes(image, decoded, width, height)
@@ -122,7 +122,7 @@ demo = gr.Interface(
     inputs=[],
     outputs=gr.Image(type="pil", label="Detected Bounding Box"),
     title="Gemma3 Object Detector",
-    description="Click 'Run' to visualize a prediction from a randomly sampled test image.",
+    description="Click 'Submit' to visualize a prediction from a randomly sampled test image.",
 )
 
 if __name__ == "__main__":