Update app.py

app.py

@@ -6,12 +6,14 @@ from transformers import AutoModelForImageSegmentation
 import torch
 from torchvision import transforms
 
-torch.set_float32_matmul_precision("high")
+# GPU 설정을 CPU로 변경
+# GPU 설정을 삭제하거나 "cuda"를 "cpu"로 변경
+# torch.set_float32_matmul_precision("high")는 CPU에선 필요 없음.
 
 birefnet = AutoModelForImageSegmentation.from_pretrained(
     "ZhengPeng7/BiRefNet", trust_remote_code=True
 )
-birefnet.to("cuda")
+birefnet.to("cpu")  # GPU -> CPU로 변경
 
 transform_image = transforms.Compose(
     [
@@ -28,10 +30,12 @@ def fn(image):
     processed_image = process(im)
     return (processed_image, origin)
 
-@spaces.GPU
+# @spaces.GPU 데코레이터 제거
+# CPU 환경에서 동작하도록 설정
+
 def process(image):
     image_size = image.size
-    input_images = transform_image(image).unsqueeze(0).to("cuda")
+    input_images = transform_image(image).unsqueeze(0).to("cpu")  # GPU -> CPU로 변경
     # Prediction
     with torch.no_grad():
         preds = birefnet(input_images)[-1].sigmoid().cpu()