rcook commited on
Commit
4dd54a9
·
verified ·
1 Parent(s): bb72e6c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import AutoTokenizer
5
  app = FastAPI()
6
 
7
  @app.get("/")
8
- def greet_json():
9
  # Example: Loading a dataset as part of the API
10
- dataset = load_dataset("squad")
11
  return {"Hello": "World!", "dataset_length": len(dataset["train"])}
 
5
  app = FastAPI()
6
 
7
  @app.get("/")
8
+ def summarize():
9
  # Example: Loading a dataset as part of the API
10
+ billsum = load_dataset("billsum", split="ca_test")
11
  return {"Hello": "World!", "dataset_length": len(dataset["train"])}