Update app.py

app.py

@@ -1017,14 +1017,10 @@ try:
     elif main_option == "AI Chat":
         prompt_user = st.text_input("Write a prompt.")
 
-        model_name = "deepseek-ai/DeepSeek-R1"
-        model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
-        tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
-
-        pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
         messages = [
             {"role": "user", "content": prompt_user},
         ]
+        pipe = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1", trust_remote_code=True)
         st.markdown(pipe(messages))
 
 except Exception as e: