update

app.py

@@ -16,6 +16,11 @@ model_path = hf_hub_download("briaai/RMBG-1.4", 'model.pth')
 if torch.cuda.is_available():
     net.load_state_dict(torch.load(model_path))
     net=net.cuda()
+    device = "cuda"
+elif torch.backends.mps.is_available():
+    net.load_state_dict(torch.load(model_path))
+    net=net.to("mps")
+    device = "mps"
 else:
     net.load_state_dict(torch.load(model_path,map_location="cpu"))
 net.eval() 
@@ -39,6 +44,9 @@ def process(image):
     im_tensor = torch.unsqueeze(im_tensor,0)
     im_tensor = torch.divide(im_tensor,255.0)
     im_tensor = normalize(im_tensor,[0.5,0.5,0.5],[1.0,1.0,1.0])
+    if device == "cuda":
+    elif device == "mps":
+        im_tensor=im_tensor.to("mps")
     if torch.cuda.is_available():
         im_tensor=im_tensor.cuda()
 
@@ -103,4 +111,4 @@ examples = [['./input.jpg'],]
 demo = gr.Interface(fn=process,inputs="image", outputs="image", examples=examples, title=title, description=description)
 
 if __name__ == "__main__":
-    demo.launch(share=False)
+    demo.launch(share=False)

requirements.txt

@@ -1,9 +1,9 @@
 gradio
 gradio_imageslider
-torch
-torchvision
+#torch
+#torchvision
 pillow
 numpy
 typing
 gitpython
-huggingface_hub
+huggingface_hub