Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update app.py
Browse files
app.py
CHANGED
@@ -63,8 +63,7 @@ app.add_middleware(
|
|
63 |
# RAG ๊ตฌ์ฑ ์์๋ฅผ ํ๋ก๊ทธ๋จ ์์ ์ ํ ๋ฒ๋ง ์ด๊ธฐํ
|
64 |
embeddings = HuggingFaceEmbeddings(
|
65 |
model_name='nlpai-lab/KURE-v1',
|
66 |
-
model_kwargs={'device': 'cpu'}
|
67 |
-
cache_folder=SENTENCE_TRANSFORMERS_HOME
|
68 |
)
|
69 |
|
70 |
try:
|
|
|
63 |
# RAG ๊ตฌ์ฑ ์์๋ฅผ ํ๋ก๊ทธ๋จ ์์ ์ ํ ๋ฒ๋ง ์ด๊ธฐํ
|
64 |
embeddings = HuggingFaceEmbeddings(
|
65 |
model_name='nlpai-lab/KURE-v1',
|
66 |
+
model_kwargs={'device': 'cpu'}
|
|
|
67 |
)
|
68 |
|
69 |
try:
|