Update app.py

app.py

@@ -56,7 +56,7 @@ def predict_keypoints(image):
     # Convert PIL → CV2 → tensor
     img = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR) / 255.0
     img_resized = cv2.resize(img, (224,224))
-    input_tensor = transform(img_resized).unsqueeze(0).to(device)
+    input_tensor = transform(img_resized).unsqueeze(0).float().to(device)
 
     with torch.no_grad():
         pred = model(input_tensor)