Update app.py

app.py

@@ -35,7 +35,7 @@ async def extract_features(file: UploadFile = File(...)):
         image = Image.open(io.BytesIO(contents)).convert("RGB")
 
         # Get feature embeddings
-        features = feature_extraction_pipeline([image])[0]
+        features = feature_extraction_pipeline(image)
         
         # Return the embedding vector
         return JSONResponse(content={"features": features})