kwak513 commited on
Commit
ad95940
ยท
verified ยท
1 Parent(s): 058dc90

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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: