Dopler47 commited on
Commit
d98b14f
·
1 Parent(s): e5cf33f

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,