Spaces:
Paused
Paused
Update app.py
Browse files
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
|
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)
|