image resize api

app.py

@@ -1,4 +1,4 @@
-from main import run_app, run_train, run_inference, run_generate_frame, run_interpolate_frames
+from main import run_app, run_train, run_inference, run_generate_frame, run_interpolate_frames, resize_images
 from evaluate import run_evaluate, get_score
 
 import spaces
@@ -33,6 +33,7 @@ with gr.Blocks() as demo:
             interp_btn = gr.Button(value="Interpolate Frames")
             eval_btn = gr.Button(value="Evaluate All")
             eval_btn2 = gr.Button(value="Evaluate")
+            resize_btn = gr.Button(value="Resize")
         with gr.Column():
             animation = gr.Video(label="Result")
             frames = gr.Gallery(type="pil", label="Frames", format="png")
@@ -68,6 +69,10 @@ with gr.Blocks() as demo:
     eval_btn2.click(
         get_score, inputs=[name, char_imgs, mocap, tr_steps, inf_steps, fps, remove_bg], outputs=[eval_scores]
     )
+
+    resize_btn.click(
+        resize_images, inputs=[frame_imgs, img_width, img_height], outputs=[frames]
+    )
     
 
 demo.launch(share=True, show_error=True)

main.py

@@ -184,8 +184,7 @@ def img_pad(img, tw, th, transparent=False):
     return new_img
 
 
-def resize_and_pad(img, target_img):
-    tw, th = target_img.size
+def resize_pad(img, tw, th, transparent):
     w, h = img.size
     
     if tw/th > w/h:
@@ -195,9 +194,12 @@ def resize_and_pad(img, target_img):
     
     img = img.resize((tw, th), Image.BICUBIC)
 
-    tw, th = target_img.size
+    return img_pad(img, tw, th, transparent)
 
-    return img_pad(img, tw, th)
+
+def resize_and_pad(img, target_img):
+    tw, th = target_img.size
+    return resize_pad(img, tw, th, False)
 
 
 def remove_zero_pad(image):
@@ -1329,7 +1331,10 @@ def run_interpolate_frames(frame1, frame2, times_to_interp, remove_bg):
     return results[0]
 
 
-
+def resize_images(images, width, height):
+    images = [img[0] for img in images]
+    return [resize_pad(img, width, height, True) for img in images]
+