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()
 
-# model2 = Generator(3, 1, 3)
-# model2.load_state_dict(torch.load('model2.pth', map_location=torch.device('cpu')))
-# model2.eval()
+model2 = Generator(3, 1, 3)
+model2.load_state_dict(torch.load('model2.pth', map_location=torch.device('cpu')))
+model2.eval()
 
 def predict(input_img, ver):
  input_img = Image.open(input_img)
@@ -99,12 +99,12 @@ def predict(input_img, ver):
 
  drawing = 0
  with torch.no_grad():
- drawing = model1(input_img)[0].detach()
+ drawing = model1(input_img)[0].detach()
  
  drawing = transforms.ToPILImage()(drawing)
  return drawing
 
-# title="Image to Coloring Page Generator"
+title="Image to Coloring Page Generator"
 # examples=[
 # ['01.jpeg', 'Complex Lines'], 
 #]