debisoft commited on
Commit
be91871
·
1 Parent(s): 0f852e5
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -164,8 +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()
171
 
 
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
+ return response
 
168
 
169
  transform2 = transforms.ToPILImage()
170