debisoft commited on
Commit
f7cf287
·
1 Parent(s): 656b0fe
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -164,7 +164,7 @@ def sample_ddim(n_sample, n=20):
164
  def greet(input):
165
  samples, intermediate = sample_ddim(32, n=1)
166
  response = transform2(transform(np.moveaxis(samples.detach().cpu().numpy(),1,3)[-1]))
167
- response2 = transform2(transform(np.moveaxis(samples.detach().cpu().numpy(),1,3)[1]))
168
  return response, response2
169
 
170
  transform2 = transforms.ToPILImage()
 
164
  def greet(input):
165
  samples, intermediate = sample_ddim(32, n=1)
166
  response = transform2(transform(np.moveaxis(samples.detach().cpu().numpy(),1,3)[-1]))
167
+ response2 = transform2(transform(np.moveaxis(samples.detach().cpu().numpy(),1,3)[1]))
168
  return response, response2
169
 
170
  transform2 = transforms.ToPILImage()