Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -58,8 +58,7 @@ class OrcaChatBot:
|
|
58 |
|
59 |
Orca_bot = OrcaChatBot(model, tokenizer)
|
60 |
|
61 |
-
def gradio_predict(user_message,
|
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 |
|