BiRefNet_for_text_writing

Running on T4

ysharma HF staff commited on 5 days ago

Commit

506dc65

•

1 Parent(s): b174f09

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -84,7 +84,7 @@ usage_to_weights_file = {
     'General-legacy': 'BiRefNet-legacy'
 }
-birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/BiRefNet_lite'), trust_remote_code=True)
 birefnet.to(device)
 birefnet.eval()

     'General-legacy': 'BiRefNet-legacy'
 }
+birefnet = AutoModelForImageSegmentation.from_pretrained('zhengpeng7/BiRefNet_lite', trust_remote_code=True)
 birefnet.to(device)
 birefnet.eval()