Tonic commited on
Commit
37c2762
1 Parent(s): 4e972fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ class IntelChatBot:
29
  prompt = f"### System:\n {self.system_message}\n ### User:\n{user_message}\n### System:\n"
30
  return prompt
31
 
32
- def Neuro(self, user_message, temperature, max_new_tokens, top_p, repetition_penalty, do_sample):
33
  prompt = self.format_prompt(user_message)
34
  inputs = self.tokenizer(prompt, return_tensors='pt', add_special_tokens=False)
35
  input_ids = inputs["input_ids"].to(self.model.device)
 
29
  prompt = f"### System:\n {self.system_message}\n ### User:\n{user_message}\n### System:\n"
30
  return prompt
31
 
32
+ def neuro(self, user_message, temperature, max_new_tokens, top_p, repetition_penalty, do_sample):
33
  prompt = self.format_prompt(user_message)
34
  inputs = self.tokenizer(prompt, return_tensors='pt', add_special_tokens=False)
35
  input_ids = inputs["input_ids"].to(self.model.device)