Update app.py

app.py

@@ -47,8 +47,14 @@ def segment(image: Image.Image, text: str, threshold: float, mask_threshold: flo
         return None, "❌ Please upload an image."
     
     try:
+        # Ensure inputs match model dtype
         inputs = processor(images=image, text=text.strip(), return_tensors="pt").to(device)
         
+        # Convert inputs to match model dtype
+        for key in inputs:
+            if inputs[key].dtype == torch.float32:
+                inputs[key] = inputs[key].to(model.dtype)
+        
         with torch.no_grad():
             outputs = model(**inputs)
         
@@ -147,6 +153,20 @@ with gr.Blocks(
     
     segment_btn = gr.Button("🎯 Segment", variant="primary", size="lg")
     
+    # Add some example prompts
+    gr.Examples(
+        examples=[
+            ["examples/person.jpg", "person"],
+            ["examples/car.jpg", "car"],
+            ["examples/dog.jpg", "dog"],
+            ["examples/building.jpg", "building"],
+        ],
+        inputs=[image_input, text_input],
+        outputs=[image_output, info_output],
+        fn=segment_example,
+        cache_examples=True,
+    )
+    
     # Clear button handler
     clear_btn.click(
         fn=clear_all,