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:
|