Update app.py

app.py

@@ -87,7 +87,7 @@ def predict(image):
     with torch.no_grad():
         img_emb = model2.encode_image(inputs)
         img_emb = normalized(img_emb.cpu().numpy())
-        img_emb_tensor = torch.from_numpy(img_emb).to(device)
+        img_emb_tensor = torch.from_numpy(img_emb).to(device).float()
         prediction = model(img_emb_tensor).item()
 
     result = {