Spaces:

m7n
/

openalex_mapper

Running on Zero

m7n commited on Feb 12

Commit

0516b4c

1 Parent(s): a26a6cb

removed brackets?!

Files changed (1) hide show

app.py CHANGED Viewed

@@ -114,7 +114,7 @@ def no_op_decorator(func):
 # #duration=120
 # @decorator_to_use
-@spaces.GPU()
 def create_embeddings(texts_to_embedd):
     """Create embeddings for the input texts using the loaded model."""
     return model.encode(texts_to_embedd, show_progress_bar=True, batch_size=192)

 # #duration=120
 # @decorator_to_use
+@spaces.GPU
 def create_embeddings(texts_to_embedd):
     """Create embeddings for the input texts using the loaded model."""
     return model.encode(texts_to_embedd, show_progress_bar=True, batch_size=192)