Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -60,7 +60,7 @@ class FalconChatBot:
|
|
| 60 |
filtered_history.append({"user": user_message, "assistant": assistant_message})
|
| 61 |
return filtered_history
|
| 62 |
|
| 63 |
-
def predict(self, user_message, assistant_message, history, temperature=
|
| 64 |
|
| 65 |
# Process the history to remove special commands
|
| 66 |
processed_history = self.process_history(history)
|
|
|
|
| 60 |
filtered_history.append({"user": user_message, "assistant": assistant_message})
|
| 61 |
return filtered_history
|
| 62 |
|
| 63 |
+
def predict(self, user_message, assistant_message, history, temperature=0.4, max_new_tokens=700, top_p=0.99, repetition_penalty=1.9):
|
| 64 |
|
| 65 |
# Process the history to remove special commands
|
| 66 |
processed_history = self.process_history(history)
|