aixsatoshi commited on
Commit
cca184e
·
verified ·
1 Parent(s): ce1c0e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ class StopOnTokens(StoppingCriteria):
36
 
37
  @spaces.GPU
38
  def predict(message, history, tokens, temperature, language):
39
- tag = "<" + language.lower() + ">"
40
  history_transformer_format = history + [[message, ""]]
41
  stop = StopOnTokens()
42
 
 
36
 
37
  @spaces.GPU
38
  def predict(message, history, tokens, temperature, language):
39
+ tag = "<" + language.lower() + ">:"
40
  history_transformer_format = history + [[message, ""]]
41
  stop = StopOnTokens()
42