Update module_chat.py

module_chat.py

@@ -12,7 +12,7 @@ def process(message, history):
     """Generate the model response given message and history
     """
     messages = vlm.build_messages(message, history)
-    return vlm.get_response(messages)
+    yield from vlm.stream_response(messages)
 
 #
 # User interface