Update app.py

app.py

@@ -11,7 +11,7 @@ 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')
-model = T5ConditionalGeneration.from_pretrained('deep-learning-analytics/GrammarCorrector').to(torch_device)
+model = T5ForConditionalGeneration.from_pretrained('deep-learning-analytics/GrammarCorrector').to(torch_device)
 
 def correct_grammar(input_text, num_correct_options=num_correct_options):
   batch = tokenizer([input_text], truncation=True, padding = 'max length', max_length = 64, return_tensors = 'pt').to(torch_device)