Update app.py

app.py

@@ -39,11 +39,12 @@ if pdf_obj:
     user_question = st.text_input("Haz una pregunta sobre tu PDF:")
 
     if user_question:
-        os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY
+        os.environ["HUGGINGFACEHUB_API_TOKEN"] = ""
         docs = knowledge_base.similarity_search(user_question, 3)
         # llm = ChatOpenAI(model_name='gpt-3.5-turbo')
-        llm = HuggingFaceHub(repo_id="lmsys/vicuna-7b-v1.1", model_kwargs={"temperature":0.5, "max_length":512})
+        # llm = HuggingFaceHub(repo_id="lmsys/vicuna-7b-v1.1", model_kwargs={"temperature":0.5, "max_length":512})
+        llm = HuggingFaceHub(repo_id="google/flan-t5-xxl", model_kwargs={"temperature":0.5, "max_length":512})
         chain = load_qa_chain(llm, chain_type="stuff")
         respuesta = chain.run(input_documents=docs, question=user_question)
 
-        st.write(respuesta)    
+        st.write(respuesta)