Hugging Face's logo

Spaces:
learneng-space
/
idioms
Sleeping

App Files Community
Nick Starkov commited on
Commit
349b690
·
1 Parent(s): f232d53

Bug Fix

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,8 +9,8 @@ os.environ["HF_HOME"] = "/app/cache"
9
 
10
  app = FastAPI()
11
 
12
- # Load the dataset from Hugging Face
13
- dataset = load_dataset("UCSC-Admire/idiom-SFT-dataset-561-2024-12-06_00-40-30", split="train")
14
 
15
  @app.get("/api/idioms")
16
  async def get_idioms():
 
9
 
10
  app = FastAPI()
11
 
12
+ # Load the dataset from Hugging Face, selecting only required columns
13
+ dataset = load_dataset("UCSC-Admire/idiom-SFT-dataset-561-2024-12-06_00-40-30", split="train").select_columns(["compound", "sentence", "output"])
14
 
15
  @app.get("/api/idioms")
16
  async def get_idioms():