Upload folder using huggingface_hub

app.py

@@ -12,6 +12,13 @@ net = ORMBG()
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 net.to(device)
 
+if torch.cuda.is_available():
+    net.load_state_dict(torch.load(model_path))
+    net = net.cuda()
+else:
+    net.load_state_dict(torch.load(model_path, map_location="cpu"))
+net.eval()
+
 
 def resize_image(image):
     image = image.convert("RGB")
@@ -59,7 +66,7 @@ gr.HTML(
   </p>
 """
 )
-title = "Background Removal"
+title = "Open Remove Background Model (ormbg)"
 description = r"""
 This model is a fully open-source background remover optimized for images with humans.