Hugging Face's logo

Spaces:
nicholasKluge
/
Aira-Demo
Running

App Files Community
nicholasKluge commited on
Commit
56d4b07
·
1 Parent(s): fa53f5b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModelForSequen
12
  dataset = load_dataset("nicholasKluge/instruct-aira-dataset", split='english')
13
 
14
  # convert the dataset to a pandas dataframe
15
- df = dataset['train'].to_pandas()
16
 
17
  # rename the columns
18
  df.columns = ['Prompt', 'Completion']
 
12
  dataset = load_dataset("nicholasKluge/instruct-aira-dataset", split='english')
13
 
14
  # convert the dataset to a pandas dataframe
15
+ df = dataset.to_pandas()
16
 
17
  # rename the columns
18
  df.columns = ['Prompt', 'Completion']