Spaces:

kwak513
/

hufschatbotdocker

Paused

kwak513 commited on Sep 7

Commit

ad95940

verified ·

1 Parent(s): 058dc90

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -63,8 +63,7 @@ app.add_middleware(
 # RAG 구성 요소를 프로그램 시작 시 한 번만 초기화
 embeddings = HuggingFaceEmbeddings(
     model_name='nlpai-lab/KURE-v1',
-    model_kwargs={'device': 'cpu'},
-    cache_folder=SENTENCE_TRANSFORMERS_HOME
 )
 try:

 # RAG 구성 요소를 프로그램 시작 시 한 번만 초기화
 embeddings = HuggingFaceEmbeddings(
     model_name='nlpai-lab/KURE-v1',
+    model_kwargs={'device': 'cpu'}
 )
 try: