image format

app.py

@@ -191,7 +191,7 @@ def upload_image(image, did, ip, user_id, user_name):
         timestamp=timestamp,
     )
     upload_url = upload_image_url
-    image_format = image.format if image.format else "PNG"
+    image_format = "PNG"
     mime_type = f"image/{image_format.lower()}"
     with BytesIO() as m_img:
         image.save(m_img, format=image_format)
@@ -380,6 +380,7 @@ with gr.Blocks(css=css) as WeShop:
             gr.Markdown("#### Step 1: Upload a person image for Pose Change")
             main_image_input = gr.Image(
                 type="pil",
+                format='png',
                 label="Main Image",
                 elem_classes=["image-container", "hide-buttons"]
             )
@@ -419,6 +420,7 @@ with gr.Blocks(css=css) as WeShop:
                 with gr.Column():
                     gr.Markdown("#### Step 3: Press 'Generate' to get your Pose Change result")
                     output = gr.Image(
+                        format='png',
                         label="Pose Change Result",
                         elem_classes=["image-container", "hide-buttons"],
                         interactive=False
@@ -459,9 +461,4 @@ with gr.Blocks(css=css) as WeShop:
                     elem_id="example-res-images"
                 )
     gr.HTML(load_description("assets/footer.html"))
-    main_image_input.upload(
-        fn=preprocess_image,
-        inputs=[main_image_input],
-        outputs=main_image_input
-    )
 WeShop.queue(api_open=False).launch(show_api=False)