s

app.py

@@ -192,18 +192,16 @@ def greet(input):
 	steps = int(input)
 	#samples, intermediate = sample_ddim(32, n=steps)
 	ctx = F.one_hot(torch.randint(0, 5, (32,)), 5).to(device=device).float()
-	samples, intermediate = sample_ddim_context(32, ctx)
+	samples, intermediate = sample_ddim_context(32, ctx, steps)
 	#response = transform2(transform(np.moveaxis(samples.detach().cpu().numpy(),1,3)[-1]))
 	#response2 = transform2(transform(np.moveaxis(samples.detach().cpu().numpy(),1,3)[1]))
 	#response = im.fromarray(intermediate[24][0][1]).convert("RGB")
 	response = intermediate.shape;
 	response2 = transform2(transform(np.moveaxis(intermediate,2,4)[0][0]))
-	#response3 = transform2(transform(np.moveaxis(intermediate,2,4)[int(steps/2)][0]))
-	#response4 = transform2(transform(np.moveaxis(intermediate,2,4)[int(steps/4)][0]))
+	response3 = transform2(transform(np.moveaxis(intermediate,2,4)[int(steps/2)][0]))
+	response4 = transform2(transform(np.moveaxis(intermediate,2,4)[int(steps/4)][0]))
 	response5 = transform2(transform(np.moveaxis(intermediate,2,4)[-1][0]))
 
-	response3 = response1
-	response4 = response5
 	return response, response2, response3, response4, response5
 
 transform2 = transforms.ToPILImage()