debisoft commited on
Commit
e180430
·
1 Parent(s): e18412d
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -198,10 +198,12 @@ def greet(input):
198
  #response = im.fromarray(intermediate[24][0][1]).convert("RGB")
199
  response = intermediate.shape;
200
  response2 = transform2(transform(np.moveaxis(intermediate,2,4)[0][0]))
201
- response3 = transform2(transform(np.moveaxis(intermediate,2,4)[int(steps/2)][0]))
202
- response4 = transform2(transform(np.moveaxis(intermediate,2,4)[int(steps/4)][0]))
203
  response5 = transform2(transform(np.moveaxis(intermediate,2,4)[-1][0]))
204
 
 
 
205
  return response, response2, response3, response4, response5
206
 
207
  transform2 = transforms.ToPILImage()
 
198
  #response = im.fromarray(intermediate[24][0][1]).convert("RGB")
199
  response = intermediate.shape;
200
  response2 = transform2(transform(np.moveaxis(intermediate,2,4)[0][0]))
201
+ #response3 = transform2(transform(np.moveaxis(intermediate,2,4)[int(steps/2)][0]))
202
+ #response4 = transform2(transform(np.moveaxis(intermediate,2,4)[int(steps/4)][0]))
203
  response5 = transform2(transform(np.moveaxis(intermediate,2,4)[-1][0]))
204
 
205
+ response3 = response1
206
+ response4 = response5
207
  return response, response2, response3, response4, response5
208
 
209
  transform2 = transforms.ToPILImage()