Upload app.py

app.py

@@ -35,7 +35,8 @@ def load_quantized_model(checkpoint_path):
     
     # Load the checkpoint
     checkpoint = torch.load(checkpoint_path, map_location="cpu")
-    model.load_state_dict(checkpoint["model_state_dict"])
+    # model.load_state_dict(checkpoint["model_state_dict"])
+    model.load_state_dict(checkpoint
     
     model.eval()
     return model