gradio could not display the generated images

[Mofr77]

Hi，

Thanks for your incredible work！
I encountered a problem in gradio_new.py,
When I tested your code, gradio could not display the generated image, and the word "error" appeared. There was no error message in the terminal. What is the problem?

Looking for your reply.

[suptimq]

Same here

[Jeffrey288]

I have the same problem here. I am running the gradio app through SSH, and I can open the gradio app in my local computer's browser because of SSH port forwarding, but the app is unable to load any of the images, presumably because of some network issues. The share=True in the gradio app should have provided a public address, but it only outputs a local IP to the app. I would appreciate it if someone could solve this problem, or point me to where I can run the model directly without the gradio app.

[elegant666yu]

I also have the same problem. No error is reported but I cannot generate images for viewing. The images are displayed incorrectly.

[zeyuanyin]

Update:
pip install gradio==4.44.1 works well for me to run gradio, but some deprecated functions need to be replaced.

---

Something is wrong with the Gradio app. I modified the code to save the generated images in the local space. FYI.

Original:

zero123/zero123/gradio_new.py

Lines 389 to 392 in f426883

	output_ims = []
	for x_sample in x_samples_ddim:
	x_sample = 255.0 * rearrange(x_sample.cpu().numpy(), 'c h w -> h w c')
	output_ims.append(Image.fromarray(x_sample.astype(np.uint8)))

Modified:

output_ims = [] 
for x_sample in x_samples_ddim: 
   x_sample = 255.0 * rearrange(x_sample.cpu().numpy(), 'c h w -> h w c') 
   output_ims.append(Image.fromarray(x_sample.astype(np.uint8))) 
for i, im in enumerate(output_ims):  # added
    im.save(f"output_im_{i}.png") # added