Update app.py

app.py

@@ -30,13 +30,7 @@ def load_model(model_path):
     )
     checkpoint = torch.load(model_path, map_location=torch.device("cpu"))
     model.load_state_dict(checkpoint['model_state_dict'])
-
-    # Rename keys to match the model definition
-state_dict['fc.weight'] = state_dict.pop('fc.1.weight')
-state_dict['fc.bias'] = state_dict.pop('fc.1.bias')
-
-# Load the modified state dict
-model.load_state_dict(state_dict)
+    
     model.eval()
     return model