Update app.py

app.py

@@ -93,14 +93,14 @@ def remove_alpha_channel(image: Image.Image) -> Image.Image:
         return image
 
 
-@torch.inference_mode()
+# @torch.inference_mode()
 def numpy2pytorch(imgs):
     h = torch.from_numpy(np.stack(imgs, axis=0)).float() / 127.0 - 1.0  # so that 127 must be strictly 0.0
     h = h.movedim(-1, 1)
     return h
 
 
-@torch.inference_mode()
+# @torch.inference_mode()
 def pytorch2numpy(imgs, quant=True):
     results = []
     for x in imgs:
@@ -123,7 +123,7 @@ def resize_without_crop(image, target_width, target_height):
     return np.array(resized_image)
 
 
-@torch.inference_mode()
+@spaces.GPU()
 def run_rmbg(img, sigma=0.0):
     """
     Remove background from image using BriaRMBG model.