enough with reasoning

app.py

@@ -24,8 +24,8 @@ gif_html.markdown(
 @st.cache_resource
 def load_model():
     # model_id = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
-    # model_id = "deepseek-ai/deepseek-llm-7b-chat"
-    model_id = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
+    model_id = "deepseek-ai/deepseek-llm-7b-chat"
+    # model_id = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
     
     # model_id = "deepseek-ai/DeepSeek-R1-Distill-Llama-8B"
     tokenizer = AutoTokenizer.from_pretrained(model_id)
@@ -84,10 +84,8 @@ if st.button("Generate"):
 
 
     # Generate text
-    chat = [{"role": "user", "content": prompt}]
-    formatted_prompt = tokenizer.apply_chat_template(chat, tokenize=False, add_generation_prompt=True)
 
-    inputs = tokenizer(formatted_prompt, return_tensors="pt").to(model.device)
+    inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
     with torch.no_grad():
         outputs = model.generate(   **inputs,
                                     # max_new_tokens=100,