Update app.py

app.py

@@ -12,7 +12,7 @@ from gramformer import Gramformer
 pipeline_en = pipeline(task="text-classification", model="Hello-SimpleAI/chatgpt-detector-roberta")
 
 # Initialize Gramformer
-gf = Gramformer(models=2, use_gpu=False)  # 2 = corrector
+gf = Gramformer(models=1, use_gpu=False)  # 1 = corrector
 
 # Function to predict the label and score for English text (AI Detection)
 def predict_en(text):
@@ -98,8 +98,8 @@ def paraphrase_correct_and_grammar(text):
     capitalized_text = capitalize_sentences_and_nouns(paraphrased_text)
     
     # Step 3: Grammar correction using Gramformer
-    corrected_sentences = gf.correct(capitalized_text)
-    final_text = corrected_sentences[0] if corrected_sentences else capitalized_text
+    corrected_sentences = gf.correct(capitalized_text, max_candidates=1)
+    final_text = next(iter(corrected_sentences)) if corrected_sentences else capitalized_text
     
     return final_text