Update app.py

app.py

@@ -100,6 +100,7 @@ def load_checkpoint(checkpoint_file, model, optimizer, lr):
         param_group["lr"] = lr
 
 genA = Generator().to(DEVICE)
+optim_gen = optim.Adam(list(genB.parameters()) + list(genA.parameters()), lr=LEARNING_RATE, betas=(0.5, 0.999))
 
 load_checkpoint(CHECKPOINT_GEN_A, genA, optim_gen, LEARNING_RATE)