test output features order

ui/app.py

@@ -27,6 +27,7 @@ class PlantDiseaseApp:
         self.model = self.model_loader.loadModel(self.current_modelName)
         self.flagged_predictions = []
         self.class_names = utils.get_class_names()
+        self.actualClassNames = self.model.fc3.out_features
 
     def predict(self, image, modelName, confidence_threshold):
         """
@@ -50,6 +51,7 @@ class PlantDiseaseApp:
             if modelName != self.current_modelName:
                 self.model, self = self.model_loader.loadModel(modelName)
                 self.current_modelName = modelName
+                self.class_names = self.model.fc3.out_features
 
             # Preprocess image
             tensor = preprocess_image(image).to(self.model_loader.device)
@@ -62,8 +64,8 @@ class PlantDiseaseApp:
             probs = torch.nn.functional.softmax(logits, dim=1).cpu().numpy()[0]
 
 
-            predID = probs.argmanx().item()
-            print("predicted index: " + predID)
+            predID = probs.argmax().item()
+            print("predicted index: " + str(predID))
 
             # Map to class names
             predictions = {name: float(prob) for name, prob in zip(self.class_names, probs)}