Adding spaces.GPU

app.py

@@ -126,7 +126,7 @@ def get_docs_from_parquet(parquet_urls, column, offset, limit):
     return df[column].tolist()
 
 
-# @spaces.GPU
+@spaces.GPU
 def calculate_embeddings(docs):
     return sentence_model.encode(docs, show_progress_bar=True, batch_size=32)
 
@@ -137,7 +137,7 @@ def calculate_n_neighbors_and_components(n_rows):
     return n_neighbors, n_components
 
 
-# @spaces.GPU
+@spaces.GPU
 def fit_model(docs, embeddings, n_neighbors, n_components):
     umap_model = UMAP(
         n_neighbors=n_neighbors,