Spaces:

nicholasKluge
/

Aira-Demo-Portuguese

Running

nicholasKluge commited on Dec 1, 2023

Commit

e1634a4

1 Parent(s): 24d747c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModelForSequen
 dataset = load_dataset("nicholasKluge/instruct-aira-dataset", split='portuguese')
 # convert the dataset to a pandas dataframe
-df = dataset['train'].to_pandas()
 # rename the columns
 df.columns = ['Prompt', 'Completion']

 dataset = load_dataset("nicholasKluge/instruct-aira-dataset", split='portuguese')
 # convert the dataset to a pandas dataframe
+df = dataset.to_pandas()
 # rename the columns
 df.columns = ['Prompt', 'Completion']