Update app.py

app.py

@@ -31,7 +31,7 @@ def correct_text(text, max_length, min_length, max_new_tokens, num_beams, temper
         )
     
     corrected_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return corrected_text
+    yield corrected_text
 
 def respond(prompt, max_length, min_length, max_new_tokens, num_beams, temperature, top_p):
     # Your response generation logic here