always save as png

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", format="jpeg", type="pil", interactive=False)
+                restored_image = gr.Image(label="Restored Image", format="png", 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", format="jpeg", type="pil", interactive=False)
-                enhanced_image = gr.Image(label="Enhanced Image", format="jpeg", type="pil", interactive=False)
+                origin_area_image = gr.Image(label="Origin Area Image", format="png", type="pil", interactive=False)
+                enhanced_image = gr.Image(label="Enhanced Image", format="png", 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", format="jpeg", type="pil", interactive=False)
+                generated_image = gr.Image(label="Generated Image", format="png", type="pil", interactive=False)
         
         g_btn.click(
             fn=segment_image,

segment_utils.py

@@ -24,10 +24,10 @@ def restore_result(croper, category, generated_image):
     restored_image.paste(cropped_generated_image, (croper.origin_start_x, croper.origin_start_y), cropped_square_mask_image)
 
     extension = 'png'
-    if restored_image.mode == 'RGBA':
-        extension = 'png'
-    else:
-        extension = 'jpg'
+    # if restored_image.mode == 'RGBA':
+    #     extension = 'png'
+    # else:
+    #     extension = 'jpg'
 
     path = f"output/{uuid.uuid4()}.{extension}"
     restored_image.save(path, quality=100)