ahmadmac commited on
Commit
d4f0738
·
verified ·
1 Parent(s): 3a686e2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -17,7 +17,11 @@ splitted_data = text_splitter.split_text(data)
17
  embeddings = HuggingFaceEmbeddings(model_name='sentence-transformers/all-mpnet-base-v2')
18
  #retriever = Qdrant.as_retriever()
19
  client = QdrantClient(":memory:")
20
- qdrant_vectorstore = Qdrant(client, embeddings.embed_query, collection_name="my_documents")
 
 
 
 
21
  retriever = qdrant_vectorstore.as_retriever()
22
  llm = HuggingFaceHub(repo_id="ahmadmac/Trained-T5-large", model_kwargs={"temperature": 0.5, "max_length": 512},huggingfacehub_api_token=hf_token)
23
  qna = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=retriever)
 
17
  embeddings = HuggingFaceEmbeddings(model_name='sentence-transformers/all-mpnet-base-v2')
18
  #retriever = Qdrant.as_retriever()
19
  client = QdrantClient(":memory:")
20
+ qdrant_vectorstore = Qdrant(
21
+ client,
22
+ embed_query,
23
+ collection_name="my_documents"
24
+ )
25
  retriever = qdrant_vectorstore.as_retriever()
26
  llm = HuggingFaceHub(repo_id="ahmadmac/Trained-T5-large", model_kwargs={"temperature": 0.5, "max_length": 512},huggingfacehub_api_token=hf_token)
27
  qna = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=retriever)