Update Chat_with_pdf_LLM.py

Chat_with_pdf_LLM.py

@@ -9,7 +9,7 @@ from langchain.callbacks import get_openai_callback
 import os
 from streamlit_chat import message
 
-def LLM_pdf(model = 'google/flan-t5-large'):
+def LLM_pdf(model_name = 'google/flan-t5-large'):
     # st.header("Ask your PDF 💬")
     
     # upload file
@@ -45,10 +45,10 @@ def LLM_pdf(model = 'google/flan-t5-large'):
       if user_question:
         docs = knowledge_base.similarity_search(user_question)
         
-        llm = HuggingFaceHub(repo_id="google/flan-t5-large", model_kwargs={"temperature":5,
+        llm = HuggingFaceHub(repo_id=model_name, model_kwargs={"temperature":5,
                                                       "max_length":64})
         chain = load_qa_chain(llm, chain_type="stuff")
-        response = chain.run(input_documents=docs,question=user_question
+        response = chain.run(input_documents=docs,question=user_question)
            
         #st.write(response)
         st.session_state.past.append(user_question)