change chain_type to rerank

app.py

@@ -5,9 +5,8 @@ import streamlit as st
 from PyPDF2 import PdfReader
 from langchain.text_splitter import CharacterTextSplitter
 from langchain.chains.question_answering import load_qa_chain
-from langchain_huggingface import HuggingFaceEmbeddings
+from langchain_huggingface import HuggingFaceEmbeddings, HuggingFacePipeline
 from langchain_community.vectorstores import FAISS
-from langchain_community.llms import HuggingFacePipeline
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 
 
@@ -60,7 +59,7 @@ def setup_qa_chain():
         print(f"Error loading model: {e}")
         return
     llm = HuggingFacePipeline(pipeline=pipe)
-    qa_chain = load_qa_chain(llm, chain_type="stuff")
+    qa_chain = load_qa_chain(llm, chain_type="map_rerank")
     
 # 메인 페이지 UI
 def main_page():