Tonic commited on
Commit
f9e3aeb
·
1 Parent(s): 309e13a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -58,9 +58,10 @@ class OrcaChatBot:
58
  do_sample=True
59
  )
60
 
61
- response = self.tokenizer.decode(output_ids[0], skip_special_tokens=True)
62
- self.update_conversation_history("", response)
63
- return response
 
64
  Orca_bot = OrcaChatBot(model, tokenizer)
65
 
66
  def gradio_predict(user_message, system_message, max_new_tokens, temperature, top_p, repetition_penalty):
 
58
  do_sample=True
59
  )
60
 
61
+ response = self.tokenizer.decode(output_ids[0], skip_special_tokens=True)
62
+ self.update_conversation_history("", response)
63
+ return response
64
+
65
  Orca_bot = OrcaChatBot(model, tokenizer)
66
 
67
  def gradio_predict(user_message, system_message, max_new_tokens, temperature, top_p, repetition_penalty):