Update app.py

app.py

@@ -70,7 +70,6 @@ def generate_response(rag_chain, input_text):
 
     return response
 
-### Ken 12/11/2024 ADD START
 def get_pdf(uploaded_file):
     temp_file = "./temp.pdf"
     if uploaded_file :
@@ -84,14 +83,13 @@ def get_pdf(uploaded_file):
     loader = PyPDFLoader(temp_file)
     docs = loader.load()
     return docs
-### Ken 12/11/2024 ADD END
 
 
 def main() -> None:
 
     st.title("🧠 This is a RAG Chatbot with Ollama and Langchain !!!")
 
-    st.write("The LLM model unsloth/Llama-3.2-3B-Instruct is used")
+    st.write("The LLM model Llama-3.2 is used")
     st.write("You can upload a PDF to chat with !!!")
 
     with st.sidebar:
@@ -100,9 +98,7 @@ def main() -> None:
 
     text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=100)
 
-    ### Ken 12/11/2024 ADD START
     raw_text = get_pdf(docs)
-    ### Ken 12/11/2024 ADD END
 
     #chunks = text_splitter.split_documents(docs)
     chunks = text_splitter.split_documents(raw_text)
@@ -140,7 +136,7 @@ def main() -> None:
 
     prompt = ChatPromptTemplate.from_template(prompt)
 
-    model = ChatOllama(model="unsloth/Llama-3.2-3B-Instruct")
+    model = ChatOllama(model="llama3.2:latest")
 
     rag_chain = (
         {"context": retriever|format_docs, "question": RunnablePassthrough()}
@@ -171,3 +167,4 @@ def main() -> None:
 
 if __name__ == "__main__":
     main()
+