update app

app.py

@@ -1,4 +1,5 @@
 import os
+import spaces
 import gradio as gr
 import numpy as np
 import torch
@@ -9,17 +10,6 @@ from gradio.themes import Soft
 from gradio.themes.utils import colors, fonts, sizes
 from transformers import Sam3Processor, Sam3Model
 
-# --- Handle optional 'spaces' import for local compatibility ---
-try:
-    import spaces
-except ImportError:
-    class spaces:
-        @staticmethod
-        def GPU(duration=60):
-            def decorator(func):
-                return func
-            return decorator
-
 colors.steel_blue = colors.Color(
     name="steel_blue",
     c50="#EBF3F8",
@@ -82,7 +72,6 @@ class SteelBlueTheme(Soft):
 
 steel_blue_theme = SteelBlueTheme()
 
-# --- Hardware Setup ---
 device = "cuda" if torch.cuda.is_available() else "cpu"
 print(f"Using device: {device}")
 
@@ -100,7 +89,7 @@ except Exception as e:
     model = None 
     processor = None
 
-@spaces.GPU(duration=60)
+@spaces.GPU
 def segment_image(input_image, text_prompt, threshold=0.5):
     if input_image is None:
         raise gr.Error("Please upload an image.")
@@ -167,7 +156,6 @@ with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
         gr.Markdown("Segment objects in images using **SAM3** (Segment Anything Model 3) with text prompts.")
 
         with gr.Row():
-            # Left Column: Inputs
             with gr.Column(scale=1):
                 input_image = gr.Image(label="Input Image", type="pil", height=300)
                 text_prompt = gr.Textbox(
@@ -178,15 +166,12 @@ with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
                 
                 run_button = gr.Button("Segment", variant="primary")
 
-            # Right Column: Output
             with gr.Column(scale=1.5):
-                # AnnotatedImage creates a nice overlay visualization
                 output_image = gr.AnnotatedImage(label="Segmented Output", height=400)
                 
                 with gr.Row():
                     threshold = gr.Slider(label="Confidence Threshold", minimum=0.0, maximum=1.0, value=0.4, step=0.05)
         
-        # Examples
         gr.Examples(
             examples=[
                 ["examples/cat.jpg", "cat", 0.5],