image format

app.py

@@ -162,7 +162,7 @@ def query_task(task_id, execution_id, did, ip, user_id, user_name):
 def upload_image(image, did, ip, user_id, user_name):
     if image is None:
         return None
-    image_format = image.format if image.format else "PNG"
+    image_format = "PNG"
     mime_type = f"image/{image_format.lower()}"
     timestamp = str(int(time.time()))
     signature = generate_signature(
@@ -485,6 +485,7 @@ with gr.Blocks(css=css) as WeShop:
                 with gr.Column():
                     gr.Markdown("#### Step 2: Press 'Generate' to get the result")
                     output = gr.Image(
+                        format='png',
                         label="Result",
                         elem_classes=["image-container", "hide-buttons"],
                         interactive=False
@@ -499,12 +500,14 @@ with gr.Blocks(css=css) as WeShop:
                 )
             with gr.Column():
                 main_image_middleware = gr.Image(
+                    format='png',
                     image_mode='RGBA',
                     type="pil",
                     label="Edited Image",
                     elem_classes=["middleware"]
                 )
                 mask_image_middleware = gr.Image(
+                    format='png',
                     image_mode='RGBA',
                     type="pil",
                     elem_classes=["middleware"]