Hugging Face's logo

Spaces:
dkdaniz
/
katara
Paused

App Files Community
1
Daniel Marques commited on
Commit
02aca85
•
1 Parent(s): f82125b

fix: add handle

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -40,7 +40,7 @@ class MyCustomSyncHandler(BaseCallbackHandler):
40
  def on_llm_new_token(self, token: str, **kwargs) -> None:
41
  ws = websocketClient.get()
42
 
43
- ws.receive_text(token)
44
 
45
  print(f"token: {token}")
46
 
 
40
  def on_llm_new_token(self, token: str, **kwargs) -> None:
41
  ws = websocketClient.get()
42
 
43
+ ws.send_text(f"token: {token}")
44
 
45
  print(f"token: {token}")
46