Fix an error in loading image in tab url.

app.py

@@ -13,6 +13,10 @@ from gradio_imageslider import ImageSlider
 from transformers import AutoModelForImageSegmentation
 from torchvision import transforms
 
+import requests
+from io import BytesIO
+
+
 torch.set_float32_matmul_precision('high')
 torch.jit.script = lambda f: f
 
@@ -55,11 +59,11 @@ birefnet.eval()
 
 
 @spaces.GPU
-def predict(
-    image: np.ndarray,
-    resolution: str,
-    weights_file: Optional[str]
-) -> Tuple[np.ndarray, np.ndarray]:
+def predict(image, resolution, weights_file):
+    if isinstance(image, str):
+        response = requests.get(image)
+        image_data = BytesIO(response.content)
+        image = np.array(Image.open(image_data))
     global birefnet
     # Load BiRefNet with chosen weights
     _weights_file = '/'.join(('zhengpeng7', usage_to_weights_file[weights_file] if weights_file is not None else usage_to_weights_file['General']))
@@ -138,6 +142,10 @@ tab_text = gr.Interface(
     outputs=ImageSlider(label="BiRefNet's prediction", type="pil"),
     examples=examples_url,
     api_name="text",
+    description=('Upload a URL, 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.'),
 )
 
 demo = gr.TabbedInterface(