Update app.py

app.py

@@ -60,6 +60,10 @@ output = pipe(
     jit=True,
 ).images
 
+output_images = pipe.numpy_to_pil(np.asarray(output.reshape((num_samples,) + output.shape[-3:])))
+output_images = image_grid(output_images, num_samples // 4, 4)
+output_images.save("tao/image.png")
+
 #gr.Interface.load("models/tsungtao/controlnet-mlsd-202305011046").launch()
 def infer(prompt, negative_prompt, image):
     # implement your inference function here