Update app.py

app.py

@@ -3,11 +3,11 @@ from datasets import load_dataset
 import matplotlib.pyplot as plt
 import seaborn as sns
 
-dataset = load_dataset('dar-tau/grammar-attention-maps-opt-350m')
+dataset = load_dataset('dar-tau/grammar-attention-maps-opt-350m')['train']
 
 def choose_sentence(name):
     return "Hello " + name + "!!"
 
 
-iface = gr.Interface(fn=choose_sentence, inputs="text", outputs="text")
+iface = gr.Interface(fn=choose_sentence, inputs=[gr.Dropdown(choices=dataset['text'])], outputs="text")
 iface.launch()