Hugging Face's logo

Spaces:
jayant-yadav
/
Unmask_distilBERT
Runtime error

App Files Community
jayant-yadav commited on
Commit
01c01b0
·
1 Parent(s): 035578f

load dataset

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ def greet(name):
9
  # iface.launch()
10
 
11
  data_files = {"train": "wikitext-2-raw-v1"}
12
- dataset = load_dataset("wikitext",data_dir='wikitext-2-raw-v1', split = "train")
13
 
14
  train_df = pd.DataFrame(dataset)
15
  print(train_df.head())
 
9
  # iface.launch()
10
 
11
  data_files = {"train": "wikitext-2-raw-v1"}
12
+ dataset = load_dataset("wikitext",'wikitext-2-raw-v1', split = "train")
13
 
14
  train_df = pd.DataFrame(dataset)
15
  print(train_df.head())