Update app.py

app.py

@@ -44,7 +44,7 @@ def correct_text2(text, genConfig):
     outputs = model.generate(inputs, **genConfig.to_dict())
 
     corrected_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    yield corrected_text
+    return corrected_text
 
 def respond(text, max_length, min_length, max_new_tokens, min_new_tokens, num_beams, temperature, top_p):
     config = GenerationConfig(