Update app.py

app.py

@@ -3,10 +3,9 @@ import streamlit as st
 import pickle
 from PyPDF2 import PdfReader
 from langchain.text_splitter import RecursiveCharacterTextSplitter
-from langchain_community.embeddings import HuggingFaceEmbeddings
+from langchain.embeddings import HuggingFaceEmbeddings
 from langchain.vectorstores import FAISS
-from langchain.llms import HuggingFace
-from langchain.chains.question_answering import load_qa_chain
+from transformers import pipeline
 import os
 
 # Load environment variables from .env file
@@ -52,12 +51,12 @@ def main():
         if query:
             docs = VectorStore.similarity_search(query=query, k=3)
 
-            # Use Hugging Face model for question answering
+            # Use Hugging Face pipeline for question answering
             model_name = "distilbert-base-uncased-distilled-squad"  # Example model
-            llm = HuggingFace(model_name=model_name)
-            chain = load_qa_chain(llm=llm, chain_type="stuff")
-            response = chain.run(input_documents=docs, question=query)
-            st.write(response)
+            qa_pipeline = pipeline("question-answering", model=model_name)
+            context = " ".join([doc.page_content for doc in docs])
+            result = qa_pipeline(question=query, context=context)
+            st.write(result['answer'])
 
 if __name__ == '__main__':
     main()