t

app.py

@@ -257,7 +257,8 @@ def greet(input):
 	ctx = torch.from_numpy(mtx_2d).to(device=device).float()
 
 	#samples, intermediate = sample_ddim_context(32, ctx, n=steps)
-	samples, intermediate = sample_ddpm_context(1, steps, ctx)
+	image_count = 1;
+	samples, intermediate = sample_ddpm_context(image_count, steps, ctx)
 
 	#samples, intermediate = sample_ddim(32, n=steps)
 	#ctx = F.one_hot(torch.randint(0, 5, (32,)), 5).to(device=device).float()
@@ -268,7 +269,7 @@ def greet(input):
 	#response = im.fromarray(intermediate[24][0][1]).convert("RGB")
 
 	sx_gen_store = np.moveaxis(intermediate,2,4)
-	nsx_gen_store = norm_all(sx_gen_store, sx_gen_store.shape[0], 1)
+	nsx_gen_store = norm_all(sx_gen_store, sx_gen_store.shape[0], image_count)
 
 	response = intermediate.shape;
 	response2 = transform2(transform(nsx_gen_store[-1][0]))