trashchenkov commited on
Commit
1f1620e
1 Parent(s): 4e3ae23

Update app.py

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