Update app.py

app.py

@@ -7,11 +7,11 @@ import re
 import uuid
 
 # Load model and tokenizer
-phi4_model_path = "deepseek-ai/DeepSeek-R1-Distill-Qwen-14B"
+our_model_path = "deepseek-ai/DeepSeek-R1-Distill-Qwen-14B"
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
-phi4_model = AutoModelForCausalLM.from_pretrained(phi4_model_path, device_map="auto", torch_dtype="auto")
-phi4_tokenizer = AutoTokenizer.from_pretrained(phi4_model_path)
+our_model = AutoModelForCausalLM.from_pretrained(our_model_path, device_map="auto", torch_dtype="auto")
+our_tokenizer = AutoTokenizer.from_pretrained(our_model_path)
 
 def format_math(text):
     text = re.sub(r"\[(.*?)\]", r"$$\1$$", text, flags=re.DOTALL)
@@ -29,8 +29,8 @@ def generate_response(user_message, max_tokens, temperature, top_p, history_stat
     if not user_message.strip():
         return history_state, history_state
 
-    model = phi4_model
-    tokenizer = phi4_tokenizer
+    model = our_model
+    tokenizer = our_tokenizer
     start_tag = "<|im_start|>"
     sep_tag = "<|im_sep|>"
     end_tag = "<|im_end|>"
@@ -121,8 +121,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             # INTRO TEXT MOVED HERE
             gr.Markdown(
                 """
-                Welcome to the Ramanujan Ganit R1 14B V1 Chatbot, developed by Fractal AI Research!  
+                Welcome to the Ramanujan Ganit R1 14B V1 Chatbot, developed by Fractal AI Research! 
+                
                 Our model excels at reasoning tasks in mathematics and science.  
+                
                 Try the example problems below from JEE Main 2025 or type in your own problems to see how our model breaks down complex reasoning problems.
                 """
             )