Spaces:
Running
Running
trashchenkov
commited on
Commit
•
1f1620e
1
Parent(s):
4e3ae23
Update app.py
Browse files
app.py
CHANGED
@@ -35,7 +35,7 @@ token = os.getenv('BOT')
|
|
35 |
user_id = os.getenv('CREATOR')
|
36 |
bot = telebot.TeleBot(token)
|
37 |
model_name = "sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
|
38 |
-
model_kwargs = {'device': '
|
39 |
encode_kwargs = {'normalize_embeddings': False}
|
40 |
embedding = HuggingFaceEmbeddings(model_name=model_name,
|
41 |
model_kwargs=model_kwargs,
|
|
|
35 |
user_id = os.getenv('CREATOR')
|
36 |
bot = telebot.TeleBot(token)
|
37 |
model_name = "sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
|
38 |
+
model_kwargs = {'device': 'gpu'}
|
39 |
encode_kwargs = {'normalize_embeddings': False}
|
40 |
embedding = HuggingFaceEmbeddings(model_name=model_name,
|
41 |
model_kwargs=model_kwargs,
|