Update gradio_app.py

gradio_app.py

@@ -136,9 +136,18 @@ def get_image(image1, prompt, image2, dim_steps=50, ddim_eta=1., fs=None, seed=1
     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))]
+        # transforms.ToTensor(),
+        # transforms.Normalize(mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5))]
     )
+
+    image1 = torch.from_numpy(image1).permute(2, 0, 1).float().cuda()
+    input_h, input_w = image1.shape[1:]
+    image1 = (image_tensor1 / 255. - 0.5) * 2
+
+    image2 = torch.from_numpy(image2).permute(2, 0, 1).float().cuda()
+    input_h, input_w = image2.shape[1:]
+    image2 = (image2 / 255. - 0.5) * 2
+
     
     # image1 = Image.open(file_list[2*idx]).convert('RGB')
     image_tensor1 = transform(image1).unsqueeze(1) # [c,1,h,w]