Update app.py

app.py

@@ -1,6 +1,7 @@
 from fastapi import FastAPI
 from datasets import load_dataset
 from transformers import AutoTokenizer
+from transformers import DataCollatorForSeq2Seq
 
 app = FastAPI()
 
@@ -31,8 +32,11 @@ def summarize():
         return model_inputs
 
     tokenized_billsum = billsum.map(preprocess_function, batched=True)
+
+    data_collator = DataCollatorForSeq2Seq(tokenizer=tokenizer, model=checkpoint)
     
-    return type(tokenized_billsum)
+    return
+#    return type(tokenized_billsum)
 
 """from fastapi import FastAPI
 from datasets import load_dataset