Update app.py

app.py

@@ -10,16 +10,12 @@ from accelerate import init_empty_weights, load_checkpoint_and_dispatch
 from datasets import load_dataset
 from huggingface_hub import login
 
-# --- Hugging Face Login ---
-HF_READONLY_API_KEY = os.getenv("HF_READONLY_API_KEY")
-login(token=HF_READONLY_API_KEY)
 
 # --- Constants ---
 COT_OPENING     = "<think>"
 EXPLANATION_OPENING = "<explanation>"
 LABEL_OPENING   = "<answer>"
 LABEL_CLOSING   = "</answer>"
-INPUT_FIELD     = "question"
 SYSTEM_PROMPT = """
 You are a guardian model evaluating the compliance of an agent with a list of rules.
 The rules will be provided in a numbered format, starting with a <rules> tag and ending with a </rules> tag.
@@ -52,6 +48,10 @@ def format_rules(rules):
     formatted_rules += "</rules>\n"
     return formatted_rules
 
+def format_transcript(transcript):
+    formatted_transcript = f"<transcript>\n{transcript}\n</transcript>\n"
+    return formatted_transcript
+
 def format_output(text):
     reasoning = re.search(r"<think>(.*?)</think>", text, flags=re.DOTALL)
     answer = re.search(r"<answer>(.*?)</answer>", text, flags=re.DOTALL)
@@ -66,12 +66,6 @@ def format_output(text):
         display += "Explanation:\n" + explanation.group(1).strip() + "\n\n"
     return display.strip() if display else text.strip()
 
-    
-def format_transcript(transcript):
-    formatted_transcript = f"<transcript>\n{transcript}\n</transcript>\n"
-    return formatted_transcript
-
-
 def get_message(model, input, system_prompt=SYSTEM_PROMPT, enable_thinking=True):
     message = model.apply_chat_template(system_prompt, input, enable_thinking=enable_thinking)
     return message