i

app.py

@@ -163,7 +163,7 @@ def sample_ddim(n_sample, n=20):
 
 def greet(input):
 	samples, intermediate = sample_ddim(32, n=25)
-	response = im.fromarray(samples.detach().cpu().numpy()[0][0]).convert("RGB")
+	response = im.fromarray(norm_torch(samples)).convert("RGB")
 	return response
 
 #iface = gr.Interface(fn=greet, inputs="text", outputs="text")