Update app/backend/main.py

app/backend/main.py

@@ -306,7 +306,7 @@ class ModelManager:
             confidence, predicted_class = torch.max(probs, dim=-1)
             
             # Convert to CPU and numpy
-            probs_np = probs.cpu().float().numpy()[0]
+            probs_np = probs.cpu().float().tolist()[0]
             predicted_idx = predicted_class.item()
             confidence_score = confidence.item()