Update app.py

app.py

@@ -301,16 +301,12 @@ def _process(img: Image.Image, prompt: str | BoundingBox | None, bg_prompt: str
             new_size = (int(img.width * ratio), int(img.height * ratio))
             img = img.resize(new_size, Image.LANCZOS)
 
-        # CUDA 메모리 관리 수정
-        try:
-            if torch.cuda.is_available():
-                current_device = torch.cuda.current_device()
-                with torch.cuda.device(current_device):
-                    torch.cuda.empty_cache()
-        except Exception as e:
-            print(f"CUDA memory management failed: {e}")
-
-        with torch.cuda.amp.autocast(enabled=torch.cuda.is_available()):
+        # CUDA 메모리 관리
+        if torch.cuda.is_available():
+            torch.cuda.empty_cache()
+
+        # 새로운 autocast 구문 사용
+        with torch.amp.autocast('cuda', enabled=torch.cuda.is_available()):
             mask, bbox, time_log = _gpu_process(img, prompt)
             masked_alpha = apply_mask(img, mask, defringe=True)