Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -55,7 +55,7 @@ class FalconChatBot:
|
|
| 55 |
return filtered_history
|
| 56 |
|
| 57 |
|
| 58 |
-
def predict(self, system_prompt, user_message, assistant_message, history, temperature=
|
| 59 |
|
| 60 |
# Process the history to remove special commands
|
| 61 |
processed_history = self.process_history(history)
|
|
|
|
| 55 |
return filtered_history
|
| 56 |
|
| 57 |
|
| 58 |
+
def predict(self, system_prompt, user_message, assistant_message, history, temperature=temperature, max_new_tokens=max_new_tokens, top_p=top_p, repetition_penalty=repetition_penalty):
|
| 59 |
|
| 60 |
# Process the history to remove special commands
|
| 61 |
processed_history = self.process_history(history)
|