Update app.py

add download button for output image

app.py

@@ -105,7 +105,7 @@ def predict(input_img, ver):
             drawing = model1(input_img)[0].detach()
     
     drawing = transforms.ToPILImage()(drawing)
-    return drawing
+    return drawing, drawing
 
 title="Image to Line Drawings - Complex and Simple Portraits and Landscapes"
 examples=[
@@ -122,6 +122,6 @@ examples=[
 
 iface = gr.Interface(predict, [gr.inputs.Image(type='filepath'),
     gr.inputs.Radio(['Complex Lines','Simple Lines'], type="value", default='Simple Lines', label='version')],
-    gr.outputs.Image(type="pil"), title=title,examples=examples)
+    [gr.outputs.Image(type="pil"), gr.outputs.DataURI(label="Download Image")], title=title,examples=examples)
 
-iface.launch()
+iface.launch()