From 0b7df4089e64d71617786b574b36dc92b59804e3 Mon Sep 17 00:00:00 2001 From: chan Date: Wed, 12 Mar 2025 11:45:29 +0900 Subject: [PATCH] =?UTF-8?q?=EB=AA=A8=EB=8D=B8=20=EC=88=98=EC=A0=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- docker-compose.yml | 6 +++--- main.py | 8 ++++---- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/docker-compose.yml b/docker-compose.yml index 8a1bf4d..4e54636 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -1,11 +1,11 @@ services: - rag_test: + rag_streamlit_test: build: . volumes: - .:/opt/workspace # 로컬 디렉토리와 컨테이너 디렉토리 동기화 ports: - - "8502:8501" # Streamlit 앱 포트 설정 - container_name: rag_test # 컨테이너 이름 설정 + - "8504:8501" # Streamlit 앱 포트 설정 + container_name: rag_streamlit_test # 컨테이너 이름 설정 deploy: resources: reservations: diff --git a/main.py b/main.py index d4ab8fd..f6afa7b 100644 --- a/main.py +++ b/main.py @@ -29,7 +29,7 @@ load_dotenv() EMBEDDING_MODEL = "nlpai-lab/KoE5" RERANK_MODEL = "upskyy/ko-reranker-8k" LOAD_DOCS = "data/fake_rag" -SAVE_VD = "fake_all" +SAVE_VD = "fake_all_KCDS" def embedding(): @@ -104,12 +104,12 @@ kiwi = Kiwi() @st.cache_resource def initialize_processing(): - processed_texts = load_and_process_documents(LOAD_DOCS) - docs = processed_texts try: faiss_index = load_faiss_index(SAVE_VD) + docs = list(faiss_index.docstore._dict.values()) except Exception: print("no load_faiss_index") + docs = load_and_process_documents(LOAD_DOCS) embeddings = embedding() faiss_index = FAISS.from_documents(docs, embeddings) save_faiss_index(faiss_index, SAVE_VD) @@ -136,7 +136,7 @@ def initialize_processing(): # GPT 모델 설정 -gpt_model = ChatOpenAI(model_name="gpt-4", temperature=0) +gpt_model = ChatOpenAI(model_name="gpt-4o-mini", temperature=0) st.set_page_config(page_title="RAG 테스트", page_icon="💬") st.title("RAG 테스트")