debisoft commited on
Commit
2c2c652
1 Parent(s): 14d3ba8
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -258,7 +258,7 @@ def greet(input):
258
 
259
  #samples, intermediate = sample_ddim_context(32, ctx, n=steps)
260
 
261
- #samples, intermediate = sample_ddpm_context(image_count, steps, ctx)
262
  samples, intermediate = sample_ddim_context(image_count, ctx, steps)
263
 
264
  #samples, intermediate = sample_ddim(32, n=steps)
@@ -274,6 +274,11 @@ def greet(input):
274
 
275
  response = intermediate.shape;
276
  response2 = transform2(transform(nsx_gen_store[-1][0]))
 
 
 
 
 
277
  # response3 = transform2(transform(nsx_gen_store[-1][1]))
278
  # response4 = transform2(transform(nsx_gen_store[-1][2]))
279
  # response5 = transform2(transform(nsx_gen_store[-1][3]))
@@ -313,7 +318,7 @@ def greet(input):
313
  #response5 = transform2(transform(np.moveaxis(intermediate,2,4)[-1][0]))
314
 
315
  ###return response, response2, response3, response4, response5, response6, response7, response8, response9, response10, response11, response12, response13, response14, response15, response16, response17, response18, response19, response20, response21, response22, response23, response24, response25, response26, response27, response28, response29, response30, response31, response32, response33
316
- return response2
317
 
318
  transform2 = transforms.ToPILImage()
319
 
 
258
 
259
  #samples, intermediate = sample_ddim_context(32, ctx, n=steps)
260
 
261
+ ###samples, intermediate = sample_ddpm_context(image_count, steps, ctx)
262
  samples, intermediate = sample_ddim_context(image_count, ctx, steps)
263
 
264
  #samples, intermediate = sample_ddim(32, n=steps)
 
274
 
275
  response = intermediate.shape;
276
  response2 = transform2(transform(nsx_gen_store[-1][0]))
277
+
278
+ sx_gen_store = np.moveaxis(samples,2,4)
279
+ nsx_gen_store = norm_all(sx_gen_store, sx_gen_store.shape[0], image_count)
280
+
281
+ response3 = transform2(transform(nsx_gen_store[-1][0]))
282
  # response3 = transform2(transform(nsx_gen_store[-1][1]))
283
  # response4 = transform2(transform(nsx_gen_store[-1][2]))
284
  # response5 = transform2(transform(nsx_gen_store[-1][3]))
 
318
  #response5 = transform2(transform(np.moveaxis(intermediate,2,4)[-1][0]))
319
 
320
  ###return response, response2, response3, response4, response5, response6, response7, response8, response9, response10, response11, response12, response13, response14, response15, response16, response17, response18, response19, response20, response21, response22, response23, response24, response25, response26, response27, response28, response29, response30, response31, response32, response33
321
+ return response3
322
 
323
  transform2 = transforms.ToPILImage()
324