Update app.py

app.py

@@ -13,9 +13,6 @@ st.title("AWS Well-Architected Review")
 st.write("Get recommendations for optimizing your AWS architecture.")
 
 
-if "nvidia_model" not in st.session_state:
-    st.session_state["nvidia_model"] = "nvidia/llama-3.1-nemotron-70b-instruct"
-
 if "messages" not in st.session_state:
     st.session_state.messages = [
         {"role": "system", "content": "You are an assistant that provides recommendations based on AWS Well-Architected Review best practices. Focus on the 5 pillars: Operational Excellence, Security, Reliability, Performance Efficiency, and Cost Optimization."}
@@ -34,7 +31,7 @@ if st.button("Get Recommendations"):
             with st.spinner("Generating recommendations..."):
                 # Create Nvidia completion request with conversation history
                 stream = client.chat.completions.create(
-                    model="nvidia-llama-3.1-70b-instruct",  # Nvidia model name
+                    model="nvidia/llama-3.1-nemotron-70b-instruct",  # Nvidia model name
                     messages=st.session_state.messages,  # Include all messages in the API call
                     temperature=0.5,
                     top_p=0.7,