final fix

app.py

@@ -28,7 +28,7 @@ def get_densenet121_mnist():
 # =========================
 def load_model():
     model = get_densenet121_mnist()
-    state_dict = torch.load("best_model.pth", map_location="cpu")
+    state_dict = torch.load("best_densenet_mnist.pth", map_location="cpu")
     model.load_state_dict(state_dict, strict=True)  # strict=True since arch matches exactly
     model.eval()
     return model