Spaces:
Paused
Paused
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)
|