Update app.py

app.py

@@ -4,6 +4,7 @@ from transformers import AutoTokenizer
 from transformers import DataCollatorForSeq2Seq
 import evaluate
 import numpy as np
+from transformers import AutoModelForSeq2SeqLM, Seq2SeqTrainingArguments, Seq2SeqTrainer
 
 app = FastAPI()
 
@@ -51,7 +52,9 @@ def summarize():
         result["gen_len"] = np.mean(prediction_lens)
 
         return {k: round(v, 4) for k, v in result.items()}
-    
+
+    model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint)
+        
     return data_collator
 #    return type(tokenized_billsum)