Update app.py

app.py

@@ -2,6 +2,7 @@ from fastapi import FastAPI
 from datasets import load_dataset
 from transformers import AutoTokenizer
 from transformers import DataCollatorForSeq2Seq
+from transformers import evaluate
 
 app = FastAPI()
 
@@ -34,6 +35,8 @@ def summarize():
     tokenized_billsum = billsum.map(preprocess_function, batched=True)
 
     data_collator = DataCollatorForSeq2Seq(tokenizer=tokenizer, model=checkpoint)
+   
+    rouge = evaluate.load("rouge")
     
     return data_collator
 #    return type(tokenized_billsum)