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