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)