Test better interface

app.py

@@ -38,9 +38,9 @@ def inference(img, size, type):
     run_cmd(f"rm -f {img_in_path}")
     run_cmd(f"rm -f {img_out_path}")
 
-    return [img_out]
+    return img_out
+
 
-input_image = gr.Image(type="pil", label="Input")
 upscale_type = gr.Radio(["Manga", "Anime", "General"], label="Select the type of picture you want to upscale:", value="Manga")
 upscale_size = gr.Radio(["x4", "x2"], label="Upscale by:", value="x4")
 output_image = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
@@ -49,14 +49,24 @@ title = "ESRGAN Upscaling With Custom Models"
 description = "This space uses old ESRGAN architecture to upscale images, using models made by the community."
 article = "<p><a href='https://upscale.wiki/wiki/Model_Database'>Model Database</a></p>"
 
-demo = gr.Interface(
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column():
+            input_image = gr.Image(type="pil", label="Input")
+            translate_btn = gr.Button(value="Upscale", variant="primary")
+        with gr.Column():
+            output_image = gr.Image(type="file")
+            save_btn = gr.Button(value="Save image")
+
+
+""" demo = gr.Interface(
     inference,
     inputs=[input_image, upscale_size, upscale_type],
     outputs=[output_image],
     title=title,
     description=description,
     article=article
-)
+) """
 
 demo.queue()
 demo.launch(debug=is_colab, share=is_colab, inline=is_colab)