Update app.py

app.py

@@ -61,13 +61,13 @@ if uploaded_file_1 and uploaded_file_2:
         tensor = val_transform(selected_image).unsqueeze(0)
 
         with torch.no_grad():
-            output = model.netG_A2B(tensor)
+            output = model.netG_B2A(tensor)
     else:
         selected_image = image2
         tensor = val_transform(selected_image).unsqueeze(0)
 
         with torch.no_grad():
-            output = model.netG_B2A(tensor)
+            output = model.netG_A2B(tensor)
 
 
     result_image = de_normalize(output.squeeze(0)).permute(1, 2, 0).numpy()