Upload app.py

app.py

@@ -10,7 +10,6 @@ import requests
 from io import BytesIO
 
 # Use CPU and smaller memory footprint
-device = "cpu"
 model = None
 processor = None
 
@@ -51,13 +50,14 @@ def get_embedding_from_url(image_url: str):
     except Exception as e:
         return {"success": False, "error": str(e)}
 
-# Gradio interface
+# Gradio interface with API enabled
 demo = gr.Interface(
     fn=get_embedding_from_url,
     inputs=gr.Textbox(label="Image URL", placeholder="https://example.com/image.jpg"),
     outputs=gr.JSON(label="Result"),
     title="CLIP Embedding API",
-    description="Get 512-dim CLIP embeddings for images"
+    description="Get 512-dim CLIP embeddings for images",
+    api_name="predict"
 )
 
 if __name__ == "__main__":