| from huggingface_hub import from_pretrained_keras | |
| import gradio as gr | |
| import tensorflow as tf | |
| model = from_pretrained_keras("emresvd/u304") | |
| gr.Interface( | |
| fn=lambda: model(tf.random.normal([1, 100]), training=False).numpy()[0, :, :, ::-1], | |
| inputs=[], | |
| outputs=gr.Image(label="output", type="numpy", show_download_button=True, height=64 * 8, width=64 * 8), | |
| ).launch() |