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())
|