Update app.py

app.py

@@ -40,8 +40,8 @@ print(checkpoint.keys())
 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'])
-scheduler.load_state_dict(checkpoint['scheduler_state_dict'])
+# optimizer.load_state_dict(checkpoint['optimizer_state_dict'])
+# scheduler.load_state_dict(checkpoint['scheduler_state_dict'])
 
 # If you want to resume from a specific epoch
 epoch = checkpoint['epoch']