Update app.py

app.py

@@ -296,10 +296,10 @@ def combine_with_background(foreground: Image.Image, background: Image.Image,
     result.paste(scaled_foreground, (x, y), scaled_foreground)
     return result
 
-@spaces.GPU(duration=30)
+@spaces.GPU(duration=20)  # 30초에서 20초로 감소
 def _gpu_process(img: Image.Image, prompt: str | BoundingBox | None) -> tuple[Image.Image, BoundingBox | None, list[str]]:
     try:
-        with torch.inference_mode():
+        with torch.inference_mode(), torch.amp.autocast('cuda', enabled=torch.cuda.is_available()):
             if isinstance(prompt, str):
                 bbox = gd_detect(img, prompt)
                 if not bbox:
@@ -309,10 +309,11 @@ def _gpu_process(img: Image.Image, prompt: str | BoundingBox | None) -> tuple[Im
                 
             mask = segmenter(img, bbox)
             return mask, bbox, []
-            
     except Exception as e:
         print(f"GPU process error: {str(e)}")
         raise
+    finally:
+        clear_memory()
 
 def _process(img: Image.Image, prompt: str | BoundingBox | None, bg_prompt: str | None = None, aspect_ratio: str = "1:1") -> tuple[tuple[Image.Image, Image.Image, Image.Image], gr.DownloadButton]:
     try: