t

app.py

@@ -247,17 +247,17 @@ def sample_ddpm_context(n_sample,timesteps, context, save_rate=20):
 
 def greet(input):
 	steps = int(input)
-
+	image_count = 1;
 	#ctx = F.one_hot(torch.randint(0, 5, (32,)), 5).to(device=device).float()
 
 	# hero, non-hero, food, spell, side-facing
 	one_hot_enc = np.array([1, 0, 0, 0, 0])
-	shape = (32, 5)
+	shape = (image_count, 5)
 	mtx_2d = np.ones(shape) * one_hot_enc
 	ctx = torch.from_numpy(mtx_2d).to(device=device).float()
 
 	#samples, intermediate = sample_ddim_context(32, ctx, n=steps)
-	image_count = 1;
+
 	samples, intermediate = sample_ddpm_context(image_count, steps, ctx)
 
 	#samples, intermediate = sample_ddim(32, n=steps)