adding 120 duration in GPU usage for topic modeling
Browse files
src/scripts/nlp_processing.py
CHANGED
|
@@ -7,7 +7,7 @@ from src.utils.constants import EMBEDDING_MODEL_NAME
|
|
| 7 |
embedding_model = SentenceTransformer(EMBEDDING_MODEL_NAME)
|
| 8 |
|
| 9 |
|
| 10 |
-
@spaces.GPU
|
| 11 |
def embed_splitted_docs(splitted_docs):
|
| 12 |
"""
|
| 13 |
Encode the given list of documents using the specified embedding model.
|
|
|
|
| 7 |
embedding_model = SentenceTransformer(EMBEDDING_MODEL_NAME)
|
| 8 |
|
| 9 |
|
| 10 |
+
@spaces.GPU
|
| 11 |
def embed_splitted_docs(splitted_docs):
|
| 12 |
"""
|
| 13 |
Encode the given list of documents using the specified embedding model.
|
src/scripts/topic_modeling.py
CHANGED
|
@@ -10,7 +10,7 @@ from src.utils.constants import EMBEDDING_MODEL_NAME
|
|
| 10 |
embedding_model = SentenceTransformer(EMBEDDING_MODEL_NAME)
|
| 11 |
|
| 12 |
|
| 13 |
-
@spaces.GPU()
|
| 14 |
def topic_modeling(
|
| 15 |
docs,
|
| 16 |
embeddings,
|
|
|
|
| 10 |
embedding_model = SentenceTransformer(EMBEDDING_MODEL_NAME)
|
| 11 |
|
| 12 |
|
| 13 |
+
@spaces.GPU(duration=120)
|
| 14 |
def topic_modeling(
|
| 15 |
docs,
|
| 16 |
embeddings,
|