Tonic commited on
Commit
4694c68
1 Parent(s): 1590544

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -58,8 +58,7 @@ class OrcaChatBot:
58
 
59
  Orca_bot = OrcaChatBot(model, tokenizer)
60
 
61
- def gradio_predict(user_message, character_intro, max_new_tokens, temperature, top_p, repetition_penalty):
62
- # Prepend the character introduction to the user message if provided
63
  full_message = f"{system_message}\n{user_message}" if system_message else user_message
64
  return Orca_bot.predict(full_message, temperature, max_new_tokens, top_p, repetition_penalty)
65
 
 
58
 
59
  Orca_bot = OrcaChatBot(model, tokenizer)
60
 
61
+ def gradio_predict(user_message, system_message, max_new_tokens, temperature, top_p, repetition_penalty):
 
62
  full_message = f"{system_message}\n{user_message}" if system_message else user_message
63
  return Orca_bot.predict(full_message, temperature, max_new_tokens, top_p, repetition_penalty)
64