i

app.py

@@ -162,12 +162,12 @@ def sample_ddim(n_sample, n=20):
     return samples, intermediate
 
 def greet(input):
-	samples, intermediate = sample_ddim(32, n=25)
+	samples, intermediate = sample_ddim(32, n=20)
 	#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 = transform2(transform(np.moveaxis(intermediate,2,4)[1][-1]))
-	response2 = transform2(transform(np.moveaxis(intermediate,2,4)[24][1]))
+	response = transform2(transform(np.moveaxis(intermediate,2,4)[1][0]))
+	response2 = transform2(transform(np.moveaxis(intermediate,2,4)[24][0]))
 	return response, response2
 
 transform2 = transforms.ToPILImage()