Update app.py

app.py

@@ -84,7 +84,7 @@ usage_to_weights_file = {
     'General-legacy': 'BiRefNet-legacy'
 }
 
-birefnet = AutoModelForImageSegmentation.from_pretrained('/'.join('zhengpeng7', 'BiRefNet_lite'), trust_remote_code=True)
+birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/BiRefNet_lite'), trust_remote_code=True)
 birefnet.to(device)
 birefnet.eval()
 
@@ -95,7 +95,7 @@ def predict(images):
 
     global birefnet
     # Load BiRefNet with chosen weights
-    _weights_file = '/'.join('zhengpeng7', 'BiRefNet_lite')
+    _weights_file = 'zhengpeng7/BiRefNet_lite'
     print('Using weights: {}.'.format(_weights_file))
     birefnet = AutoModelForImageSegmentation.from_pretrained(_weights_file, trust_remote_code=True)
     birefnet.to(device)