Update app.py

app.py

@@ -7,7 +7,7 @@ default_text = "This should working"
 sent = st.text_area("Text", default_text, height=40)
 num_correct_options = st.number_input('Number of Correction Options', min_value=1, max_value=3, value=1, step=1)
 
-from transformers import T5ConditionalGeneration, T5Tokenizer
+from transformers import T5ForConditionalGeneration, T5Tokenizer
 import torch
 torch_device = 'cuda' if torch.cude.is_available() else 'cpu'
 tokenizer = T5Tokenizer.from_pretrained('deep-learning-analytics/GrammarCorrector')