Update app.py

app.py

@@ -13,12 +13,10 @@ def predict(image):
             image_features = model.encode_image(image)
             image_features /= image_features.norm(dim=-1, keepdim=True)
 
-    embedding = image_features[0]
-    print(embedding)
+    embedding = image_features[0].cpu().float().numpy()
 
     model = joblib.load('model.pkl')
     result = model.predict([embedding])
-    print(result)
 
     return "Map" if result == 1 else "No map"