Tonic commited on
Commit
6ab3fad
1 Parent(s): 048bd80

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -51,7 +51,8 @@ class FalconChatBot:
51
  return filtered_history
52
 
53
 
54
- def predict(self, system_prompt, user_message, assistant_message, history, max_length=500):
 
55
  # Process the history to remove special commands
56
  processed_history = self.process_history(history)
57
 
 
51
  return filtered_history
52
 
53
 
54
+ def predict(self, system_prompt, user_message, assistant_message, history, temperature, max_new_tokens, top_p, repetition_penalty):
55
+
56
  # Process the history to remove special commands
57
  processed_history = self.process_history(history)
58