debisoft commited on
Commit
a048868
·
1 Parent(s): d594716
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -162,14 +162,14 @@ def sample_ddim(n_sample, n=20):
162
  return samples, intermediate
163
 
164
  def greet(input):
165
- samples, intermediate = sample_ddim(32, n=50)
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
  #response = im.fromarray(intermediate[24][0][1]).convert("RGB")
169
  response = intermediate.shape;
170
  response2 = transform2(transform(np.moveaxis(intermediate,2,4)[0][-1]))
171
- response3 = transform2(transform(np.moveaxis(intermediate,2,4)[10][-1]))
172
- response4 = transform2(transform(np.moveaxis(intermediate,2,4)[30][-1]))
173
  response5 = transform2(transform(np.moveaxis(intermediate,2,4)[-1][-1]))
174
 
175
  return response, response2, response3, response4, response5
 
162
  return samples, intermediate
163
 
164
  def greet(input):
165
+ samples, intermediate = sample_ddim(32, n=100)
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
  #response = im.fromarray(intermediate[24][0][1]).convert("RGB")
169
  response = intermediate.shape;
170
  response2 = transform2(transform(np.moveaxis(intermediate,2,4)[0][-1]))
171
+ response3 = transform2(transform(np.moveaxis(intermediate,2,4)[50][-1]))
172
+ response4 = transform2(transform(np.moveaxis(intermediate,2,4)[70][-1]))
173
  response5 = transform2(transform(np.moveaxis(intermediate,2,4)[-1][-1]))
174
 
175
  return response, response2, response3, response4, response5