IMage ot be downlaoded

app.py

@@ -341,8 +341,17 @@ def pose_transfer(source_image, test_pair_index):
         sampling_imgs = sampling_imgs.clamp(0, 1)
     
     # Convert to PIL image
-    output_img = Image.fromarray((sampling_imgs[0] * 255.).permute((1, 2, 0)).long().cpu().numpy().astype(np.uint8)).resize((256, 256))
-    
+    # output_img = Image.fromarray((sampling_imgs[0] * 255.).permute((1, 2, 0)).long().cpu().numpy().astype(np.uint8)).resize((256, 256))
+    # Convert tensor to PIL image without resizing
+    output_img = Image.fromarray(
+        (sampling_imgs[0] * 255.)
+        .permute((1, 2, 0))
+        .long()
+        .cpu()
+        .numpy()
+        .astype(np.uint8)
+    )
+
     return output_img
 
 def start_download():
@@ -440,7 +449,14 @@ with gr.Blocks() as demo:
                 generate_btn = gr.Button("🚀 Generate Pose Transfer")
             
             with gr.Column():
-                output_image = gr.Image(label="Output Image")
+                # output_image = gr.Image(label="Output Image")
+                output_image = gr.Image(
+                    label="Output Image",
+                    type="pil",                  # ✅ Must be 'pil'
+                    format="png",                # ✅ Force PNG encoding
+                    show_download_button=True    # ✅ Enables download
+                )
+
                 status_message = gr.Textbox(label="Status", interactive=False)
         
         generate_btn.click(