Fix embedding funtion

app.py

@@ -33,11 +33,10 @@ transform = transforms.Compose([
 ])
 
 # Function to extract embedding
-def get_embedding(image_path):
-    image = Image.open(image_path).convert('RGB')
-    image = transform(image).unsqueeze(0)  # Add batch dimension
+def get_embedding(image):
+    image = transform(image).unsqueeze(0)
     with torch.no_grad():
-        embedding = model(image).squeeze()  # Remove extra dimensions
+        embedding = model(image).squeeze()
     return embedding.numpy()
 
 # Prediction function