Update app.py

app.py

@@ -183,7 +183,7 @@ model = ModelManager(
     )
 
 image_type = 'pil' # filepath' 
-def predict(input, img_enhancer):  
+def predict(input, platform_radio, img_enhancer):  
     if input is None:
         return None, [], gr.update(visible=False)
     if image_type == 'filepath':
@@ -199,7 +199,10 @@ def predict(input, img_enhancer):
         image = np.array(image_pil)
         mask = np.array(mask_pil.convert("L"))
     output, ori_image = model_process(image, mask, img_enhancer)
-    return output, [resize_image(ori_image, new_width=400), resize_image(output, new_width=400)], gr.update(visible=True)
+    if platform_radio == 'pc':
+        return output, [ori_image, output], gr.update(visible=True)
+    else:
+        return output, [resize_image(ori_image, new_width=400), resize_image(output, new_width=400)], gr.update(visible=True)
 
 image_blocks = gr.Blocks(css=css, title='Image Cleaner')
 with image_blocks as demo:
@@ -256,6 +259,8 @@ with image_blocks as demo:
                     gallery = gr.Gallery(
                             label="Generated images", show_label=False, elem_id="gallery"
                         ).style(grid=[2], height="600px")
+                    platform_radio = gr.Radio(["pc", "mobile"],  elem_id="platform_radio",value="pc", 
+                                    label="platform:", show_label=True, visible=False)                    
             with gr.Row(elem_id="download-container", visible=False) as download_container:             
                 with gr.Column(elem_id="download-btn-container") as download_btn_container:
                     download_button = gr.Button(elem_id="download-btn", value="Save(⏩)")  
@@ -273,7 +278,7 @@ with image_blocks as demo:
                     """
                     ) 
 
-            erase_btn.click(fn=predict, inputs=[image_input, img_enhancer], outputs=[image_out, gallery, download_container])
+            erase_btn.click(fn=predict, inputs=[image_input, platform_radio, img_enhancer], outputs=[image_out, gallery, download_container])
             download_button.click(None, [], [], _js=download_img)
             share_button.click(None, [], [], _js=share_js)