Update app.py

app.py

@@ -9,7 +9,7 @@ if not HF_API_TOKEN:
     raise ValueError("Hugging Face API Token is not set in the environment variables.")
 
 # Hugging Face API URL and header for Gemma 27B-it model
-GEMMA_27B_API_URL = "https://api-inference.huggingface.co/models/google/gemma-2-27b-it"
+GEMMA_9B_API_URL = "https://api-inference.huggingface.co/models/google/gemma-2-9b-it"
 
 HEADERS = {"Authorization": f"Bearer {HF_API_TOKEN}"}
 
@@ -28,9 +28,9 @@ def extract_pdf_text(uploaded_file):
     return pdf_text
 
 # Streamlit app
-st.set_page_config(page_title="Gemma 27B-it Chatbot Interface", layout="wide")
-st.title("Gemma 27B-it Chatbot Interface")
-st.write("Gemma 27B-it Chatbot Interface")
+st.set_page_config(page_title="Gemma 9B-it Chatbot Interface", layout="wide")
+st.title("Gemma 9B-it Chatbot Interface")
+st.write("Gemma 9B-it Chatbot Interface")
 
 # Initialize session state for conversation and uploaded file
 if "conversation" not in st.session_state:
@@ -52,9 +52,9 @@ question = st.text_input("Ask a question...", "")
 if st.button("Send") and question:
     try:
         with st.spinner("Waiting for the model to respond..."):
-            response = query_model(GEMMA_27B_API_URL, {"inputs": question})
+            response = query_model(GEMMA_9B_API_URL, {"inputs": question})
             answer = response.get("generated_text", "No response")
-            st.write(f"**Gemma 27B-it:** {answer}")
+            st.write(f"**Gemma 9B-it:** {answer}")
             st.session_state.conversation.append((question, answer))
     except ValueError as e:
         st.error(str(e))
@@ -93,5 +93,5 @@ if st.session_state.conversation:
     st.write('<div class="chat-container">', unsafe_allow_html=True)
     for user_message, bot_message in st.session_state.conversation:
         st.write(f'<div class="user-message">You: {user_message}</div>', unsafe_allow_html=True)
-        st.write(f'<div class="bot-message">Gemma 27B-it: {bot_message}</div>', unsafe_allow_html=True)
+        st.write(f'<div class="bot-message">Gemma 9B-it: {bot_message}</div>', unsafe_allow_html=True)
     st.write('</div>', unsafe_allow_html=True)