Update app.py

app.py

@@ -119,15 +119,17 @@ class ModelHandler:
 
         return self._format_predictions(ensemble_probs)
 
-    def _format_predictions(self, probs):
+        def _format_predictions(self, probs):
         top5_probs, top5_indices = torch.topk(probs, 5)
         result = {}
         for i in range(5):
             idx = top5_indices[0][i].item()
             label = label_mapping.get(idx, f"Класс {idx}")
-            result[label] = float(top5_probs[0][i].item() * 100)
+            # return raw prob, not percent:
+            result[label] = float(top5_probs[0][i].item())
         return result
 
+
 # Initialize model handler
 model_handler = ModelHandler()