Update app.py

app.py

@@ -41,7 +41,7 @@ from transformers import AutoTokenizer
 app = FastAPI()
 
 # Load dataset and tokenizer
-billsum = load_dataset("billsum", split="train[:1%]")  # Load a small sample
+billsum = load_dataset("billsum", split="ca_test")  # Load a small sample
 tokenizer = AutoTokenizer.from_pretrained("t5-small")
 prefix = "summarize: "  # Example prefix for text generation