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 |
|