Update inference.py

inference.py

@@ -19,7 +19,7 @@ class Inferencer:
             transforms.Normalize((0.1307,), (0.3081,))
         ])
         
-    def _load_model(self, model_path='saved_models/best_model.pth'):
+    def _load_model(self, model_path='best_model.pth'):
         """Load the trained model."""
         model = MNISTModel().to(self.device)
         model.load_state_dict(