reverting to original width & height

app.py

@@ -121,10 +121,10 @@ pipe = TryonPipeline.from_pretrained(
 )
 pipe.unet_encoder = UNet_Encoder
 
-WIDTH = 4160 # 768
-HEIGHT = 6240 # 1024
-POSE_WIDTH = int(WIDTH/8)
-POSE_HEIGHT = int(HEIGHT/8)
+WIDTH = 768  # 4160 # 768
+HEIGHT = 1024  # 6240 # 1024
+POSE_WIDTH = int(WIDTH/2)  # int(WIDTH/8)
+POSE_HEIGHT = int(HEIGHT/2)  #int(HEIGHT/8)
 
 @spaces.GPU
 def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_steps,seed):
@@ -292,10 +292,10 @@ with image_blocks as demo:
                 examples=garm_list_path)
         with gr.Column():
             # image_out = gr.Image(label="Output", elem_id="output-img", height=400)
-            masked_img = gr.Image(label="Masked image output", elem_id="masked-img",show_share_button=False)
+            masked_img = gr.Image(label="Masked image output", elem_id="masked-img", show_share_button=False)
         with gr.Column():
             # image_out = gr.Image(label="Output", elem_id="output-img", height=400)
-            image_out = gr.Image(label="Output", elem_id="output-img",show_share_button=False)
+            image_out = gr.Image(label="Output", elem_id="output-img", show_share_button=False)
 
 
 

requirements.txt

@@ -20,4 +20,8 @@ av
 fvcore
 cloudpickle
 omegaconf
-pycocotools
+pycocotools
+
+# For Local
+#gradio==4.24.0
+#spaces==0.28.3