Update app.py

app.py

@@ -24,9 +24,11 @@ class BiRefNetDummy(torch.nn.Module):
         # Will be replaced by actual model code in the next step.
         return torch.ones_like(x)
 
-def load_model():
-    # Replace with actual class once copied from the BiRefNet repo!
-    net = BiRefNetDummy()
+ddef load_model():
+    download_weights()
+    net = BiRefNet()
+    net.load_state_dict(torch.load("BiRefNet.pth", map_location="cpu"))
+    net.eval()
     return net
 
 bi_ref_net = load_model()