Update app/ui.py

app/ui.py

@@ -1,5 +1,5 @@
 from gen.openllm import (
-    chat,
+    gen_chat,
     translate_messages,
 )
 
@@ -8,6 +8,6 @@ async def chat(message, history):
     messages.append({"role":"user", "content":message})
     
     full_resp = ""
-    async for resp in chat(messages, history):
+    async for resp in gen_chat(messages):
         full_resp = full_resp + resp
         yield full_resp