update script

app.py

@@ -22,8 +22,10 @@ transform_image = transforms.Compose(
 
 
 @spaces.GPU
-def rmbg(image):
-    image = load_img().convert("RGB")
+def rmbg(image,url):
+    if image is None : 
+        image = url
+    image = load_img(image).convert("RGB")
     image_size = image.size
     input_images = transform_image(image).unsqueeze(0).to("cuda")
     # Prediction
@@ -36,7 +38,7 @@ def rmbg(image):
     return image
 
 
-rmbg_tab = gr.Interface(fn=rmbg, inputs=["text"], outputs=["image"], api_name="rmbg")
+rmbg_tab = gr.Interface(fn=rmbg, inputs=["image","url"], outputs=["image"], api_name="rmbg")
 
 demo = gr.TabbedInterface(
     [rmbg_tab],