celalkartoglu commited on
Commit
e7e9ed3
·
verified ·
1 Parent(s): 36503b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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', max_length=100)
 
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())