Update app.py

app.py

@@ -87,7 +87,7 @@ class ModelManager:
                 'ConvNextV2Base_best_model.pth',
                 map_location=self.device
             )
-            self._breed_model.load_state_dict(checkpoint['base_model'], strict=False)
+            self._breed_model.load_state_dict(checkpoint['model_state_dict'], strict=False)
             self._breed_model.eval()
         return self._breed_model