Hugging Face's logo

Spaces:
bhardwaj08sarthak
/
STEM-Question-Generator
Running

App Files Community
bhardwaj08sarthak commited on
Commit
b087291
·
verified ·
1 Parent(s): 0afce5c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -57,7 +57,7 @@ DOK_PHRASES = {
57
  }
58
 
59
  # Prebuild embeddings once
60
- _backend = HFEmbeddingBackend(model_name="sentence-transformers/all-MiniLM-L6-v2")
61
  _BLOOM_INDEX = build_phrase_index(_backend, BLOOMS_PHRASES)
62
  _DOK_INDEX = build_phrase_index(_backend, DOK_PHRASES)
63
 
 
57
  }
58
 
59
  # Prebuild embeddings once
60
+ _backend = HFEmbeddingBackend(model_name="sentence-transformers/all-MiniLM-L6-v2").to('cuda')
61
  _BLOOM_INDEX = build_phrase_index(_backend, BLOOMS_PHRASES)
62
  _DOK_INDEX = build_phrase_index(_backend, DOK_PHRASES)
63