Tonic commited on
Commit
3ea1758
1 Parent(s): 5ff99f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=temperature, max_new_tokens=max_new_tokens, top_p=top_p, repetition_penalty=repetition_penalty):
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)