Update app.py

update app to only compute complex lines

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,29 +99,19 @@ def predict(input_img, ver):
 
     drawing = 0
     with torch.no_grad():
-        if ver == 'Simple Lines':
-            drawing = model2(input_img)[0].detach()
-        else:
             drawing = model1(input_img)[0].detach()
     
     drawing = transforms.ToPILImage()(drawing)
     return drawing
 
-title="Image to Line Drawings - Complex and Simple Portraits and Landscapes"
-examples=[
-['01.jpeg', 'Simple Lines'], ['02.jpeg', 'Simple Lines'], ['03.jpeg', 'Simple Lines'],
-['07.jpeg', 'Complex Lines'], ['08.jpeg', 'Complex Lines'], ['09.jpeg', 'Complex Lines'],
-['10.jpeg', 'Simple Lines'], ['11.jpeg', 'Simple Lines'], ['12.jpeg', 'Simple Lines'],
-['01.jpeg', 'Complex Lines'], ['02.jpeg', 'Complex Lines'], ['03.jpeg', 'Complex Lines'],
-['04.jpeg', 'Simple Lines'], ['05.jpeg', 'Simple Lines'], ['06.jpeg', 'Simple Lines'],
-['07.jpeg', 'Simple Lines'], ['08.jpeg', 'Simple Lines'], ['09.jpeg', 'Simple Lines'],
-['04.jpeg', 'Complex Lines'], ['05.jpeg', 'Complex Lines'], ['06.jpeg', 'Complex Lines'],
-['10.jpeg', 'Complex Lines'], ['11.jpeg', 'Complex Lines'], ['12.jpeg', 'Complex Lines'],
-['Upload Wild Horses 2.jpeg', 'Complex Lines']
+# title="Image to Coloring Page Generator"
+# examples=[
+# ['01.jpeg', 'Complex Lines'], 
+
 ]
 
 iface = gr.Interface(predict, [gr.inputs.Image(type='filepath'),
-    gr.inputs.Radio(['Complex Lines','Simple Lines'], type="value", default='Simple Lines', label='version')],
+    gr.inputs.Radio(['Complex Lines'], type="value", default='Complex Lines', label='version')],
     gr.outputs.Image(type="pil"), title=title,examples=examples)
 
 iface.launch()