Update app.py

app.py

@@ -22,7 +22,8 @@ def qa_from_pdf(pdf_file, question):
     embedding_model = HuggingFaceEmbeddings(model_name='sentence-transformers/all-MiniLM-L6-v2')
     db = FAISS.from_documents(documents, embedding_model)
 
-    hf_pipeline = pipeline('text-generation', model='tiiuae/falcon-rw-1b', max_length=512)
+    hf_pipeline = pipeline('text-generation', model='sshleifer/tiny-gpt2', max_length=100)
+
     llm = HuggingFacePipeline(pipeline=hf_pipeline)
 
     qa_chain = RetrievalQA.from_chain_type(llm=llm, retriever=db.as_retriever())