Moha782 commited on
Commit
68c3755
·
verified ·
1 Parent(s): 02ae9d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from huggingface_hub import InferenceClient
3
  from langchain_community.document_loaders import PyPDFLoader
4
  from langchain.text_splitter import RecursiveCharacterTextSplitter
5
- from langchain_community.vectorstores import FAISS
6
  from sentence_transformers import SentenceTransformer
7
  from langchain.chains import RetrievalQA
8
  from langchain_community.llms import HuggingFaceHub
@@ -20,7 +20,7 @@ embeddings = SentenceTransformer("sentence-transformers/all-mpnet-base-v2")
20
  texts = [doc.page_content for doc in texts] # Get the text content from the documents
21
  embeddings = embeddings.encode(texts) # Get the embeddings for the texts
22
 
23
- vector_store = FAISS.from_texts_and_embeddings(texts, embeddings)
24
 
25
  # Initialize the HuggingFaceHub LLM
26
  llm = HuggingFaceHub(repo_id="HuggingFaceH4/zephyr-7b-beta", model_kwargs={"temperature": None, "top_p": None})
 
2
  from huggingface_hub import InferenceClient
3
  from langchain_community.document_loaders import PyPDFLoader
4
  from langchain.text_splitter import RecursiveCharacterTextSplitter
5
+ from langchain_community.vectorstores.faiss import FAISS
6
  from sentence_transformers import SentenceTransformer
7
  from langchain.chains import RetrievalQA
8
  from langchain_community.llms import HuggingFaceHub
 
20
  texts = [doc.page_content for doc in texts] # Get the text content from the documents
21
  embeddings = embeddings.encode(texts) # Get the embeddings for the texts
22
 
23
+ vector_store = FAISS.from_texts(texts, embeddings)
24
 
25
  # Initialize the HuggingFaceHub LLM
26
  llm = HuggingFaceHub(repo_id="HuggingFaceH4/zephyr-7b-beta", model_kwargs={"temperature": None, "top_p": None})