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