app.py

import gradio as gr

def show_sample(i):
    return dataset[i]["image"]

gr.Interface(
    fn=show_sample,
    inputs=gr.Number(value=0, precision=0),
    outputs=gr.Image(type="pil")
).launch()