t

app.py

@@ -168,10 +168,10 @@ def greet(input):
 	#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][-1]))
-	response3 = transform2(transform(np.moveaxis(intermediate,2,4)[int(steps/2)][-1]))
-	response4 = transform2(transform(np.moveaxis(intermediate,2,4)[int(steps/4)][-1]))
-	response5 = transform2(transform(np.moveaxis(intermediate,2,4)[-1][-1]))
+	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]))
+	response5 = transform2(transform(np.moveaxis(intermediate,2,4)[-1][0]))
 
 	return response, response2, response3, response4, response5