Adding zero scaling :)

backend/semantic_search.py

@@ -4,6 +4,7 @@ import time
 
 import lancedb
 from sentence_transformers import SentenceTransformer
+import spaces
 
 
 # Setting up the logging
@@ -28,6 +29,7 @@ name="sentence-transformers/paraphrase-multilingual-minilm-l12-v2"
 st_model = SentenceTransformer(name, device='cuda')
 
 # used for both training and querying
+@spaces.GPU
 def embed_func(query):
     return st_model.encode(query)