import gradio as gr
from datasets import load_dataset
import matplotlib.pyplot as plt
import seaborn as sns

dataset = load_dataset('dar-tau/grammar-attention-maps-opt-350m')

def choose_sentence(name):
    return "Hello " + name + "!!"


iface = gr.Interface(fn=choose_sentence, inputs="text", outputs="text")
iface.launch()