add image preprocessing

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import os
 import time
 import requests
+from PIL import Image
 
 
 # ux format
@@ -67,6 +68,13 @@ def load_header(header_file):
         content = f.read()
     return content
 
+def preprocess_img(img_path, max_size=1024):
+    img = Image.open(img_path)
+    if max(img.size) > max_size:
+        img.thumbnail((max_size, max_size))
+        img.save(img_path)
+    return img_path
+
 def update_category(selected_garm_file):
     selected_category = garm_img_category_mapping.get(os.path.basename(selected_garm_file), "Fullbody")
     return gr.update(value=selected_category)
@@ -195,6 +203,9 @@ with gr.Blocks(css=tryon_css) as Huhu_Turbo:
         inputs=[garm_img],
         outputs=[category]
     )
+    
+    garm_img.change(fn=preprocess_img, inputs=[garm_img], outputs=[garm_img])
+    person_img.change(fn=preprocess_img, inputs=[person_img], outputs=[person_img])
 
     generate_button.click(fn=run_turbo, inputs=[person_img, garm_img, category], outputs=[result_img, result_info], api_name=False, concurrency_limit=30)
 

requirements.txt

@@ -0,0 +1 @@
+Pillow