load dataset

app.py

@@ -8,9 +8,8 @@ def greet(name):
 # iface = gr.Interface(fn=greet, inputs="text", outputs="text")
 # iface.launch()
 
-data_files = {"train": "train.csv"}
-dataset = load_dataset("lhoestq/demo1", split = "train")
-print(dataset)
+data_files = {"train": "wikitext-2-raw-v1"}
+dataset = load_dataset("wikitext", data_files= data_files, split = "train")
 
 train_df = pd.DataFrame(dataset)
 print(train_df.loc[4,'review'])