wangzhang commited on
Commit
f2eea57
·
1 Parent(s): b8f99e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ def inference(message, history):
9
  # Translate to English
10
  message = translator.translate(message, dest="en").text
11
  partial_message = ""
12
- for token in client.text_generation(message, max_new_tokens=1024, stream=True):
13
  partial_message += token
14
  yield partial_message
15
 
 
9
  # Translate to English
10
  message = translator.translate(message, dest="en").text
11
  partial_message = ""
12
+ for token in client.text_generation(str(message), max_new_tokens=1024, stream=True):
13
  partial_message += token
14
  yield partial_message
15