Update app.py

app.py

@@ -185,6 +185,7 @@ def calculate_dimensions(aspect_ratio: str, base_size: int = 512) -> tuple[int,
         return base_size * 4 // 3, base_size
     return base_size, base_size
 
+@spaces.GPU(duration=40)  # 배경 생성은 30초 제한
 def generate_background(prompt: str, aspect_ratio: str) -> Image.Image:
     try:
         width, height = calculate_dimensions(aspect_ratio)
@@ -217,7 +218,6 @@ def generate_background(prompt: str, aspect_ratio: str) -> Image.Image:
         print(f"Background generation error: {str(e)}")
         return Image.new('RGB', (512, 512), 'white')
 
-
 def create_position_grid():
     return """
     <div class="position-grid" style="display: grid; grid-template-columns: repeat(3, 1fr); gap: 10px; width: 150px; margin: auto;">
@@ -274,22 +274,26 @@ def combine_with_background(foreground: Image.Image, background: Image.Image,
     result.paste(scaled_foreground, (x, y), scaled_foreground)
     return result
 
-@spaces.GPU
+@spaces.GPU(duration=120)  # 60초 동안만 GPU 리소스 사용
 def _gpu_process(img: Image.Image, prompt: str | BoundingBox | None) -> tuple[Image.Image, BoundingBox | None, list[str]]:
     time_log: list[str] = []
-    if isinstance(prompt, str):
+    try:
+        if isinstance(prompt, str):
+            t0 = time.time()
+            bbox = gd_detect(img, prompt)
+            time_log.append(f"detect: {time.time() - t0}")
+            if not bbox:
+                print(time_log[0])
+                raise gr.Error("No object detected")
+        else:
+            bbox = prompt
         t0 = time.time()
-        bbox = gd_detect(img, prompt)
-        time_log.append(f"detect: {time.time() - t0}")
-        if not bbox:
-            print(time_log[0])
-            raise gr.Error("No object detected")
-    else:
-        bbox = prompt
-    t0 = time.time()
-    mask = segmenter(img, bbox)
-    time_log.append(f"segment: {time.time() - t0}")
-    return mask, bbox, time_log
+        mask = segmenter(img, bbox)
+        time_log.append(f"segment: {time.time() - t0}")
+        return mask, bbox, time_log
+    except Exception as e:
+        print(f"GPU process error: {str(e)}")
+        raise
 
 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: