Update app.py

app.py

@@ -2,14 +2,14 @@ import gradio as gr
 from transformers import AutoTokenizer, T5ForConditionalGeneration
 
 # Load the CoEdIT-xl model and tokenizer
-tokenizer = AutoTokenizer.from_pretrained("grammarly/coedit-xl")
-model = T5ForConditionalGeneration.from_pretrained("grammarly/coedit-xl")
+tokenizer = AutoTokenizer.from_pretrained("grammarly/coedit-xxl")
+model = T5ForConditionalGeneration.from_pretrained("grammarly/coedit-xxl")
 
 def edit_text(input_text):
     # Tokenize input text
     input_ids = tokenizer(input_text, return_tensors="pt").input_ids
     # Generate edited text
-    outputs = model.generate(input_ids, max_length=256)
+    outputs = model.generate(input_ids, max_length=1005)
     edited_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return edited_text