Update app.py

app.py

@@ -47,7 +47,8 @@ def predict(input_image):
         logits = model(x)
         probs = F.softmax(logits, dim=1)  # convert to probabilities
         
-    results = {cls: float(prob) for cls, prob in zip(CLASSES, probs)}
+    probs = probs.squeeze(0).cpu().numpy().tolist()  # ensures it's a Python list of floats
+    results = {cls: prob for cls, prob in zip(CLASSES, probs)}
     
     return results