Update

app.py

@@ -1,12 +1,12 @@
 import streamlit as st
-from transformers import (AutoModel, AutoTokenizer, pipeline)
+from transformers import (AutoModel, AutoTokenizer, pipeline, T5ForConditionalGeneration)
 
 # import torch
 
 st.set_page_config(layout="wide", page_title="تبدیل متون محاوره‌ای به فارسی")
 
 model_path = 'erfan226/persian-t5-formality-transfer'
-model = AutoModel.from_pretrained(model_path)
+model = T5ForConditionalGeneration.from_pretrained(model_path)
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 pipe = pipeline(task='text2text-generation', model=model, tokenizer=tokenizer)