Small change

app.py

@@ -1,8 +1,6 @@
 import numpy as np
 import gradio as gr
 import segment_anything
-import imutils
-import numpy as np
 import base64
 import torch
 import typing
@@ -11,12 +9,10 @@ import subprocess
 
 
 def image_to_sam_image_embedding(
-    image_url: str,
+    image: np.ndarray,
     model_size: typing.Literal["base", "large", "huge"] = "base",
 ) -> str:
     """Generate an image embedding."""
-    # Load image
-    image = imutils.url_to_image(image_url)
 
     # Select model size
     if model_size == "base":
@@ -80,7 +76,12 @@ if __name__ == "__main__":
     # Gradio app
     app = gr.Interface(
         fn=image_to_sam_image_embedding,
-        inputs="text",
+        inputs=[
+            gr.inputs.Image(type="url", label="Image URL"),
+            gr.inputs.Radio(
+                choices=["base", "large", "huge"], label="Model Size", default="base"
+            ),
+        ],
         outputs="text",
     )
     app.launch()

requirements.txt

@@ -1,7 +1,6 @@
 numpy==1.19.4
 torch==1.13.0
 torchvision
-imutils
 git+https://github.com/facebookresearch/segment-anything.git
 pycocotools
 onnxruntime