Update app.py

app.py

@@ -7,19 +7,22 @@ spell_corrector = pipeline("text2text-generation", model=model_id, max_length=10
 
 
 
-def correct_spell(text):
+
+def correct_spell(text, num_sequence=3):
     d = Differ()
     if text is None:
         text = ""
-    corrected = spell_corrector(text)[0]['generated_text']
-
+    # corrected = spell_corrector(text, num_return_sequences=3)[0]['generated_text']
+    corrected = spell_corrector(text,num_beams=5, num_return_sequences=num_sequence)
+    first_suggestion = corrected[0]['generated_text']
+    corrected = "\n\n".join([c['generated_text'] for c in corrected])
+    
     return [
         (token[2:], token[0] if token[0] != " " else None)
-        for token in d.compare(text, corrected)
+        for token in d.compare(text, first_suggestion)
     ], corrected
 
 
-
 demo = gr.Interface(
     correct_spell,
     [