Update app.py

app.py

@@ -1,11 +1,11 @@
 import streamlit as st
 text = st.text_area('You is here')
 ### Run Model
-from transformers import T5ForConditionalGeneration, T5Tokenizer, AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import T5ForConditionalGeneration, T5Tokenizer
 import torch
 torch_device = 'cuda' if torch.cuda.is_available() else 'cpu'
-tokenizer = AutoTokenizer.from_pretrained('deep-learning-analytics/GrammarCorrector')
-model = AutoModelForSeq2SeqLM.from_pretrained('deep-learning-analytics/GrammarCorrector').to(torch_device)
+tokenizer = T5Tokenizer.from_pretrained('deep-learning-analytics/GrammarCorrector')
+model = T5ForConditionalGeneration.from_pretrained('deep-learning-analytics/GrammarCorrector').to(torch_device)
 
 def correct_grammar(input_text,num_return_sequences=1):
   batch = tokenizer([input_text],truncation=True,padding='max_length',max_length=64, return_tensors="pt").to(torch_device)