fbnnb commited on
Commit
a55e17b
Β·
verified Β·
1 Parent(s): 74b4608

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. 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:]