CodCodingCode commited on
Commit
6a2015d
·
1 Parent(s): 0576ce3

fixed treatment input code

Browse files
Files changed (1) hide show
  1. app.py +1 -7
app.py CHANGED
@@ -65,16 +65,10 @@ print(
65
  # === Role Agent with instruction/input/output format ===
66
  class RoleAgent:
67
  def __init__(self, role_instruction, tokenizer, model):
68
- format_instr = (
69
- "When you respond, ALWAYS use exactly two sections:\n"
70
- "THINKING: <your detailed reasoning>\n"
71
- "ANSWER: <your concise final output>\n"
72
- "Do NOT output anything else.\n\n"
73
- )
74
 
75
  self.tokenizer = tokenizer
76
  self.model = model
77
- self.role_instruction = role_instruction + format_instr
78
 
79
  def act(self, input_text):
80
  prompt = (
 
65
  # === Role Agent with instruction/input/output format ===
66
  class RoleAgent:
67
  def __init__(self, role_instruction, tokenizer, model):
 
 
 
 
 
 
68
 
69
  self.tokenizer = tokenizer
70
  self.model = model
71
+ self.role_instruction = role_instruction
72
 
73
  def act(self, input_text):
74
  prompt = (