Update module_rewriting.py

module_rewriting.py

@@ -76,7 +76,8 @@ def rewrite_text(text, instruction, tgt_lang):
             ]
         }
     ]    
-    return vlm.get_response(messages)
+    yield from vlm.stream_response(messages)
+
 
 #
 # User interface