add image format

app_base.py

@@ -95,13 +95,13 @@ def create_demo() -> gr.Blocks:
             with gr.Column():
                 input_image = gr.Image(label="Input Image", type="pil")
             with gr.Column():
-                restored_image = gr.Image(label="Restored Image", type="pil", interactive=False)
+                restored_image = gr.Image(label="Restored Image", format="jpeg", type="pil", interactive=False)
                 download_path = gr.File(label="Download the output image", interactive=False)
             with gr.Column():
-                origin_area_image = gr.Image(label="Origin Area Image", type="pil", interactive=False)
-                enhanced_image = gr.Image(label="Enhanced Image", type="pil", interactive=False)
+                origin_area_image = gr.Image(label="Origin Area Image", format="jpeg", type="pil", interactive=False)
+                enhanced_image = gr.Image(label="Enhanced Image", format="jpeg", type="pil", interactive=False)
                 generated_cost = gr.Textbox(label="Time cost by step (ms):", visible=True, interactive=False)
-                generated_image = gr.Image(label="Generated Image", type="pil", interactive=False)
+                generated_image = gr.Image(label="Generated Image", format="jpeg", type="pil", interactive=False)
         
         g_btn.click(
             fn=segment_image,