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()