greyfoss commited on
Commit
5171bfd
1 Parent(s): ee072dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -36,7 +36,6 @@ def format(history):
36
  else:
37
  prompt += f"{bot_token}{txt}"
38
  prompt += bot_token
39
- logger.info(prompt)
40
  return prompt
41
 
42
  def remove_spaces_between_chinese(text):
@@ -82,7 +81,8 @@ def gradio(model, tokenizer):
82
  output = beam_output[0][prompt_length:]
83
 
84
  generated = remove_spaces_between_chinese(tokenizer.decode(output, skip_special_tokens=True, clean_up_tokenization_spaces=True))
85
-
 
86
  return generated
87
 
88
  bot = gr.Chatbot(show_copy_button=True, show_share_button=True, height="2000")
 
36
  else:
37
  prompt += f"{bot_token}{txt}"
38
  prompt += bot_token
 
39
  return prompt
40
 
41
  def remove_spaces_between_chinese(text):
 
81
  output = beam_output[0][prompt_length:]
82
 
83
  generated = remove_spaces_between_chinese(tokenizer.decode(output, skip_special_tokens=True, clean_up_tokenization_spaces=True))
84
+ logger.info(prompt+generated)
85
+
86
  return generated
87
 
88
  bot = gr.Chatbot(show_copy_button=True, show_share_button=True, height="2000")