Update app.py

app.py

@@ -95,8 +95,8 @@ model3.eval()
 # model2.load_state_dict(torch.load('model2.pth', map_location=torch.device('cpu')))
 # model2.eval()
 
-def predict(input_img):
-    input_img = Image.open(input_img, ver)
+def predict(input_img, ver):
+    input_img = Image.open(input_img)
     transform = transforms.Compose([transforms.Resize(256, Image.BICUBIC), transforms.ToTensor()])
     input_img = transform(input_img)
     input_img = torch.unsqueeze(input_img, 0)
@@ -123,7 +123,7 @@ title="Image to Coloring Page Generator"
 
 
 iface = gr.Interface(predict, [gr.inputs.Image(type='filepath'),
-    gr.inputs.Radio(['Complex Lines'], type="value", default='Complex Lines', label='version')],
+    gr.inputs.Radio(['Complex Lines','Simple Lines'], type="value", default='Complex Lines', label='version')],
     gr.outputs.Image(type="pil"))
 
 iface.launch()