Update app.py

app.py

@@ -3,11 +3,10 @@ import streamlit as st
 import pickle
 from PyPDF2 import PdfReader
 from langchain.text_splitter import RecursiveCharacterTextSplitter
-from langchain.embeddings import HuggingFaceEmbeddings
+from langchain_community.embeddings import HuggingFaceEmbeddings
 from langchain.vectorstores import FAISS
 from langchain.llms import HuggingFace
 from langchain.chains.question_answering import load_qa_chain
-from langchain.callbacks import get_openai_callback
 import os
 
 # Load environment variables from .env file
@@ -57,9 +56,7 @@ def main():
             model_name = "distilbert-base-uncased-distilled-squad"  # Example model
             llm = HuggingFace(model_name=model_name)
             chain = load_qa_chain(llm=llm, chain_type="stuff")
-            with get_openai_callback() as cb:
-                response = chain.run(input_documents=docs, question=query)
-                print(cb)
+            response = chain.run(input_documents=docs, question=query)
             st.write(response)
 
 if __name__ == '__main__':