updates to app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
 import torch
-from transformers import AutoTokenizer, AutoModelWithLMHead
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 tokenizer = AutoTokenizer.from_pretrained("mmcquade11/autonlp-reuters-summarization-34018133")
-model = AutoModelWithLMHead.from_pretrained("mmcquade11/autonlp-reuters-summarization-34018133")
+model = AutoModelForSeq2SeqLM.from_pretrained("mmcquade11/autonlp-reuters-summarization-34018133")
 
 def summarize(text):
     input_ids = torch.tensor(tokenizer.encode(text, add_special_tokens=True)).unsqueeze(0)