Add weights option to BiRefNet trained in all different settings.

app.py

@@ -57,7 +57,9 @@ def predict(image, resolution, weights_file):
     global birefnet
     if weights_file != weights_path:
         # Load BiRefNet with chosen weights
-        birefnet = AutoModelForImageSegmentation.from_pretrained('/'.join(('zhengpeng7', usage_to_weights_file[weights_file] if weights_file is not None else 'BiRefNet')), trust_remote_code=True)
+        _weights_file = '/'.join(('zhengpeng7', usage_to_weights_file[weights_file] if weights_file is not None else 'BiRefNet'))
+        print('Change weights to:', _weights_file)
+        birefnet = AutoModelForImageSegmentation.from_pretrained(_weights_file, trust_remote_code=True)
         birefnet.to(device)
         birefnet.eval()
         weights_path = weights_file