Update gradio_app.py
Browse files- gradio_app.py +2 -2
gradio_app.py
CHANGED
@@ -137,8 +137,8 @@ def get_image(image1, prompt, image2, dim_steps=50, ddim_eta=1., fs=None, seed=1
|
|
137 |
transforms.Resize(min(video_size)),
|
138 |
transforms.CenterCrop(video_size),
|
139 |
# transforms.ToTensor(),
|
140 |
-
# transforms.Normalize(mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5))
|
141 |
-
)
|
142 |
|
143 |
image1 = torch.from_numpy(image1).permute(2, 0, 1).float().cuda()
|
144 |
input_h, input_w = image1.shape[1:]
|
|
|
137 |
transforms.Resize(min(video_size)),
|
138 |
transforms.CenterCrop(video_size),
|
139 |
# transforms.ToTensor(),
|
140 |
+
# transforms.Normalize(mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5))
|
141 |
+
])
|
142 |
|
143 |
image1 = torch.from_numpy(image1).permute(2, 0, 1).float().cuda()
|
144 |
input_h, input_w = image1.shape[1:]
|