Create app.py

app.py

@@ -0,0 +1,29 @@
+from fastapi import FastAPI, UploadFile, File
+from pydantic import BaseModel
+from sentence_transformers import SentenceTransformer
+from PIL import Image
+import io
+
+app = FastAPI()
+
+# Load model into memory (do this globally so it only happens once)
+model = SentenceTransformer('google/siglip-so400m-patch14-384')
+
+class TextRequest(BaseModel):
+    text: str
+
+@app.post("/embed-text")
+async def embed_text(request: TextRequest):
+    # Convert text to vector
+    vector = model.encode(request.text).tolist()
+    return {"vector": vector}
+
+@app.post("/embed-image")
+async def embed_image(file: UploadFile = File(...)):
+    # Read uploaded image
+    image_data = await file.read()
+    image = Image.open(io.BytesIO(image_data))
+    
+    # Convert image to vector
+    vector = model.encode(image).tolist()
+    return {"vector": vector}