Tonic commited on
Commit
434332b
1 Parent(s): 1baccf7

Update app.py

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