Update app.py

app.py

@@ -10,6 +10,7 @@ if not HF_API_TOKEN:
 # Hugging Face API URLs and headers for Gemma models
 GEMMA_7B_API_URL = "https://api-inference.huggingface.co/models/google/gemma-1.1-7b-it"
 GEMMA_27B_API_URL = "https://api-inference.huggingface.co/models/google/gemma-2-27b-it"
+CODEGEMMA_7B_API_URL = "https://api-inference.huggingface.co/models/google/codegemma-7b"
 
 HEADERS = {"Authorization": f"Bearer {HF_API_TOKEN}"}
 
@@ -29,10 +30,10 @@ st.write("Gemma Chatbot Interface")
 if "conversation" not in st.session_state:
     st.session_state.conversation = []
 if "model_history" not in st.session_state:
-    st.session_state.model_history = {model: [] for model in ["Gemma-1.1-7B", "Gemma-2-27B"]}
+    st.session_state.model_history = {model: [] for model in ["Gemma-1.1-7B", "Gemma-2-27B", "codegemma-7b"]}
 
 # Dropdown for Gemma model selection
-gemma_selection = st.selectbox("Select Gemma Model", ["Gemma-1.1-7B", "Gemma-2-27B"])
+gemma_selection = st.selectbox("Select Gemma Model", ["Gemma-1.1-7B", "Gemma-2-27B", "codegemma-7b"])
 
 # User input for question
 question = st.text_input("Question", placeholder="Enter your question here...")
@@ -46,6 +47,8 @@ if st.button("Send") and question:
                 response = query_model(GEMMA_7B_API_URL, {"inputs": chat_history})
             elif gemma_selection == "Gemma-2-27B":
                 response = query_model(GEMMA_27B_API_URL, {"inputs": chat_history})
+            elif gemma_selection == "codegemma-7b":
+                response = query_model(CODEGEMMA_7B_API_URL, {"inputs": chat_history})
 
             answer = response.get("generated_text", "No response") if isinstance(response, dict) else response[0].get("generated_text", "No response") if isinstance(response, list) else "No response"
             add_message_to_conversation(question, answer, gemma_selection)