Update app.py

app.py

@@ -87,9 +87,9 @@ model1 = Generator(3, 1, 3)
 model1.load_state_dict(torch.load('model.pth', map_location=torch.device('cpu')))
 model1.eval()
 
-model3 = Generator(3, 1, 3)
-model3.load_state_dict(torch.load('model.pth', map_location=torch.device('cpu')))
-model3.eval()
+# model3 = Generator(3, 1, 3)
+# model3.load_state_dict(torch.load('model.pth', map_location=torch.device('cpu')))
+# model3.eval()
 
 # model2 = Generator(3, 1, 3)
 # model2.load_state_dict(torch.load('model2.pth', map_location=torch.device('cpu')))
@@ -103,7 +103,7 @@ def predict(input_img):
 
     drawing = 0
     with torch.no_grad():
-        drawing = model3(input_img)[0].detach()
+        drawing = model1(input_img)[0].detach()
     
     drawing = transforms.ToPILImage()(drawing)
     return drawing