Update app.py

app.py

@@ -12,7 +12,7 @@ app = FastAPI()
 def summarize():
     # Example: Loading a dataset as part of the API
     billsum = load_dataset("billsum", split="ca_test")
-
+    billsum = billsum.train_test_split(test_size=0.2)
 #    import pandas as pd
 
 #    df = pd.read_csv("squad_sample_train.tsv", sep="\t")