from datasets import load_dataset
import gradio as gr
import pandas as pd


dataset = load_dataset("ammaan/countrytable")

train_data = dataset['train']

dataset_list = []
for row in train_data:
    dataset_list.append(dict(row))


def data_table():
    df = pd.DataFrame(dataset_list)
    html_table = df.to_html(index=False)
    return html_table

demo = gr.Interface(
    fn=data_table,
    inputs=None,
    outputs='html',
)

demo.launch()