Update app.py

app.py

@@ -7,8 +7,6 @@ import torch
 import spaces
 from transformers import AutoModelForCausalLM, AutoTokenizer, AutoConfig
 from accelerate import init_empty_weights, load_checkpoint_and_dispatch
-from datasets import load_dataset
-from huggingface_hub import login
 
 
 # --- Constants ---
@@ -66,10 +64,6 @@ def format_output(text):
         display += "Explanation:\n" + explanation.group(1).strip() + "\n\n"
     return display.strip() if display else text.strip()
 
-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
-
 # --- Model Handling ---
 class ModelWrapper:
     def __init__(self, model_name):