t

app.py

@@ -163,7 +163,7 @@ def sample_ddim(n_sample, n=20):
 
 def greet(input):
 	samples, intermediate = sample_ddim(32, n=1)
-	response = transform2(transform(np.moveaxis(samples.detach().cpu().numpy(),1,3)[0]))
+	response = transform2(transform(np.moveaxis(samples.detach().cpu().numpy(),1,3)[-1]))
 	return response
 
 transform2 = transforms.ToPILImage()