load dataset

app.py

@@ -9,9 +9,10 @@ def greet(name):
 # iface.launch()
 
 data_files = {"train": "wikitext-2-raw-v1"}
-dataset = load_dataset("wikitext", split = "train")
+dataset = load_dataset("wikitext",'wikitext-2-raw-v1', split = "train")
 
 train_df = pd.DataFrame(dataset)
 print(train_df.head())
+print(train_df.dtypes)
 # data_files = {"train": "train.csv", "test": "test.csv"}
 # dataset = load_dataset("namespace/your_dataset_name", data_files=data_files)