Update app.py

app.py

@@ -1020,7 +1020,7 @@ try:
         messages = [
             {"role": "user", "content": prompt_user},
         ]
-        pipe = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1", trust_remote_code=True)
+        pipe = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1", trust_remote_code=True, device_map="auto") # Added device_map
         st.markdown(pipe(messages))
 
 except Exception as e: