Update app.py

app.py

@@ -120,14 +120,14 @@ class ModelHandler:
         return self._format_predictions(ensemble_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}")
-            # return raw prob, not percent:
-            result[label] = float(top5_probs[0][i].item())
-        return result
+            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}")
+                # return raw prob, not percent:
+                result[label] = float(top5_probs[0][i].item())
+            return result
 
 
 # Initialize model handler