update app.py

app.py

@@ -1,13 +1,13 @@
 from transformers import pipeline
 import gradio as gr
 
-from transformers import AutoTokenizer, AutoModelWithLMHead
+from transformers import AutoTokenizer, AutoModelForCausalLM
 #from os import path
 
 #MODEL_DIRECTORY = "/models/mrm8488-t5-base-finetuned-emotion"
 tokenizer = AutoTokenizer.from_pretrained("tuner007/pegasus_paraphrase", use_fast=False)
 #if not path.exists(MODEL_DIRECTORY):
-model = AutoModelWithLMHead.from_pretrained("tuner007/pegasus_paraphrase")
+model = AutoModelForCausalLM.from_pretrained("tuner007/pegasus_paraphrase")
 #    model.save_pretrained(MODEL_DIRECTORY)
 #else:
 #    model = AutoModelWithLMHead.from_pretrained(MODEL_DIRECTORY)