Update app.py

app.py

@@ -11,7 +11,7 @@ torch.set_float32_matmul_precision(["high", "highest"][0])
 birefnet = AutoModelForImageSegmentation.from_pretrained(
     "ZhengPeng7/BiRefNet", trust_remote_code=True
 )
-birefnet.to("cuda")
+# birefnet.to("cuda")
 
 transform_image = transforms.Compose(
     [
@@ -28,7 +28,7 @@ def fn(image):
     processed_image = process(im)
     return (processed_image, origin)
 
-@spaces.GPU
+# @spaces.GPU
 def process(image):
     image_size = image.size
     input_images = transform_image(image).unsqueeze(0).to("cuda")