Updated app

app.py

@@ -20,7 +20,7 @@ st.set_page_config(layout="wide")
 def load_model():
     tokenizer = AutoTokenizer.from_pretrained(cfg["model_name_or_path"])
     model = GPT2LMHeadModel.from_pretrained(cfg["model_name_or_path"])
-    generator = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
+    generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
     return generator, tokenizer
 
 with st.spinner('Loading model...'):