Update app.py

app.py

@@ -1,9 +1,7 @@
 import base64
-
 import streamlit as st
 from streamlit_chat import message
 from streamlit_extras.colored_header import colored_header
-
 from backend import QnASystem
 from schema import TransformType, EmbeddingTypes, IndexerType, BotType
 
@@ -14,7 +12,6 @@ st.set_page_config(page_title="PDFChat - An LLM-powered experimentation app")
 if "qna_system" not in st.session_state:
     st.session_state.qna_system = QnASystem()
 
-
 def show_pdf(f):
     f.seek(0)
     base64_pdf = base64.b64encode(f.read()).decode('utf-8')
@@ -22,12 +19,10 @@ def show_pdf(f):
                   f'type="application/pdf"></iframe>'
     st.markdown(pdf_display, unsafe_allow_html=True)
 
-
 def model_settings():
     kwargs["temperature"] = st.slider("Temperature", max_value=1.0, min_value=0.0)
     kwargs["max_tokens"] = st.number_input("Max Token", min_value=0, value=512)
 
-
 st.title("PDF Question and Answering")
 
 tab1, tab2, tab3 = st.tabs(["Upload and Ingest PDF", "Ask", "Show PDF"])
@@ -53,7 +48,7 @@ with st.sidebar:
             api_key = st.text_input("Hugging Face API Key", placeholder="hg-...", type="password")
             if not api_key.startswith('hg-'):
                 st.warning('Please enter your HuggingFace API key!', icon='⚠')
-            huggingface_model = st.selectbox("Choose Model", options=["google/flan-t5-xl", "huggingface/model2", "huggingface/model3"])
+            kwargs["model_name"] = st.selectbox("Choose Model", options=["google/flan-t5-xl"])
             model_settings()
         case EmbeddingTypes.COHERE:
             api_key = st.text_input("Cohere API Key", placeholder="...", type="password")
@@ -92,14 +87,12 @@ with tab1:
                 st.session_state.qna_system.build_chain(transform_type=text_transform, embedding_type=selected_model,
                                                         indexer_type=vector_indexer, **kwargs)
 
-
 def generate_response(prompt):
     if prompt and uploaded_file:
         response = st.session_state.qna_system.ask_question(prompt)
         return response.get("answer", response.get("result", "")), response.get("source_documents")
     return "", []
 
-
 with tab2:
     if not uploaded_file:
         st.warning("Please upload PDF", icon='⚠')
@@ -128,12 +121,10 @@ with tab2:
                 response_container = st.container()
                 response = ""
 
-
                 def get_text():
                     input_text = st.text_input("You: ", "", key="input")
                     return input_text
 
-
                 with input_container:
                     user_input = get_text()
                     if st.button("Clear"):
@@ -168,4 +159,4 @@ with tab3:
     elif uploaded_file:
         st.warning("Feature not enabled.", icon='⚠')
     else:
-        st.warning("Please upload PDF", icon='⚠')
+        st.warning("Please upload PDF", icon='⚠')