changed invalid argument in model load

app.py

@@ -37,7 +37,7 @@ checkpoint = torch.load('model.pth', map_location='cpu')
 print(checkpoint.keys())
 
 # Load the model weights
-model.load_state_dict(checkpoint['model'], strict=False, map_location=device)
+model.load_state_dict(checkpoint['model'], strict=False)
 
 # If you need to resume training, load optimizer and scheduler states
 optimizer.load_state_dict(checkpoint['optimizer_state_dict'])