Update app.py

app.py

@@ -11,17 +11,13 @@ from transformers import (
     StoppingCriteriaList
 )
 
-MODEL_ID ="Daemontatox/Phi-4-COT"
+MODEL_ID ="Daemontatox/Cogito-R1"
 
 
 # 
 # 
 
 DEFAULT_SYSTEM_PROMPT ="""
-
-You are a deep thinking AI, you may use extremely long chains of thought to deeply consider the problem and deliberate with yourself via systematic reasoning processes to help come to a correct solution prior to answering. You should enclose your thoughts and internal monologue inside <think> </think> tags, and then provide your solution or response to the problem.
-
-
 """
 
 CSS = """
@@ -48,7 +44,7 @@ def initialize_model():
     model = AutoModelForCausalLM.from_pretrained(
         MODEL_ID,
         device_map="cuda",
-        #quantization_config=quantization_config,
+        quantization_config=quantization_config,
         torch_dtype=torch.bfloat16,
         trust_remote_code=True
     )