Hugging Face's logo

Spaces:
pratyushmaini
/
tofu_viewer
Running

App Files Community
pratyushmaini commited on
Commit
053e250
•
1 Parent(s): 2a75753

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
3
  from datasets import load_dataset
4
 
5
  tofu_ds = load_dataset('locuslab/TOFU', "full")
6
- ds = tofu_ds
7
 
8
  df = ds.to_pandas()
9
  cols = list(df.columns)
 
3
  from datasets import load_dataset
4
 
5
  tofu_ds = load_dataset('locuslab/TOFU', "full")
6
+ ds = tofu_ds["train"]
7
 
8
  df = ds.to_pandas()
9
  cols = list(df.columns)