Update gradio_app.py

gradio_app.py

@@ -132,6 +132,14 @@ def get_image(image1, prompt, image2, dim_steps=50, ddim_eta=1., fs=None, seed=1
                         loop=False, interp=False, timestep_spacing='uniform', guidance_rescale=0.0, noise_shape=[72, 108], n_samples=1, **kwargs):
     
     seed_everything(seed)
+
+    transform = transforms.Compose([
+        transforms.Resize(min(video_size)),
+        transforms.CenterCrop(video_size),
+        transforms.ToTensor(),
+        transforms.Normalize(mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5))]
+    )
+    
     # image1 = Image.open(file_list[2*idx]).convert('RGB')
     image_tensor1 = transform(image1).unsqueeze(1) # [c,1,h,w]
     # image2 = Image.open(file_list[2*idx+1]).convert('RGB')