Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,8 @@ def qa_from_pdf(pdf_path, question):
|
|
17 |
embedding_model = HuggingFaceEmbeddings(model_name='sentence-transformers/all-MiniLM-L6-v2')
|
18 |
db = FAISS.from_documents(documents, embedding_model)
|
19 |
|
20 |
-
hf_pipeline = pipeline('text-generation', model='sshleifer/tiny-gpt2',
|
|
|
21 |
llm = HuggingFacePipeline(pipeline=hf_pipeline)
|
22 |
|
23 |
qa_chain = RetrievalQA.from_chain_type(llm=llm, retriever=db.as_retriever())
|
|
|
17 |
embedding_model = HuggingFaceEmbeddings(model_name='sentence-transformers/all-MiniLM-L6-v2')
|
18 |
db = FAISS.from_documents(documents, embedding_model)
|
19 |
|
20 |
+
hf_pipeline = pipeline('text-generation', model='sshleifer/tiny-gpt2', max_new_tokens=100)
|
21 |
+
|
22 |
llm = HuggingFacePipeline(pipeline=hf_pipeline)
|
23 |
|
24 |
qa_chain = RetrievalQA.from_chain_type(llm=llm, retriever=db.as_retriever())
|