Add an additional tab for inference with url.

app.py

@@ -107,17 +107,35 @@ for idx_example, example in enumerate(examples):
 examples.append(examples[-1].copy())
 examples[-1][1] = '512x512'
 
-demo = gr.Interface(
+tab_image = gr.Interface(
  fn=predict,
  inputs=[
- 'image',
+ gr.Image(label='Upload an image'),
  gr.Textbox(lines=1, placeholder="Type the resolution (`WxH`) you want, e.g., `1024x1024`. Higher resolutions can be much slower for inference.", label="Resolution"),
  gr.Radio(list(usage_to_weights_file.keys()), value='General', label="Weights", info="Choose the weights you want.")
  ],
- outputs=ImageSlider(),
+ outputs=ImageSlider(label="BiRefNet's prediction", type="pil"),
  examples=examples,
- title='Online demo for `Bilateral Reference for High-Resolution Dichotomous Image Segmentation`',
- description=('Upload a picture, our model will extract a highly accurate segmentation of the subject in it. :)'
- '\nThe resolution used in our training was `1024x1024`, thus the suggested resolution to obtain good results!\n Ours codes can be found at https://github.com/ZhengPeng7/BiRefNet.\n We also maintain the HF model of BiRefNet at https://huggingface.co/ZhengPeng7/BiRefNet for easier access.')
+ api_name="image"
 )
-demo.launch(debug=True)
+
+tab_text = gr.Interface(
+ fn=predict,
+ inputs=gr.Textbox(label="Paste an image URL"),
+ outputs=ImageSlider(label="BiRefNet's prediction", type="pil"),
+ examples=["https://hips.hearstapps.com/hmg-prod/images/gettyimages-1229892983-square.jpg"],
+ api_name="text"
+)
+
+demo = gr.TabbedInterface(
+ [tab_image, tab_text],
+ ["image", "text"],
+ title="BiRefNet demo for subject extraction (general / salient / camouflaged / portrait)."
+ description=('Upload a picture, our model will extract a highly accurate segmentation of the subject in it.\n)'
+ ' The resolution used in our training was `1024x1024`, thus the suggested resolution to obtain good results!\n'
+ ' Our codes can be found at https://github.com/ZhengPeng7/BiRefNet.\n'
+ ' We also maintain the HF model of BiRefNet at https://huggingface.co/ZhengPeng7/BiRefNet for easier access.')
+)
+
+if __name__ == "__main__":
+ demo.launch(debug=True)