Enable spaces GPU

app.py

@@ -26,7 +26,7 @@ from sentence_transformers import SentenceTransformer
 from dotenv import load_dotenv
 import os
 
-# import spaces
+import spaces
 import gradio as gr
 
 
@@ -151,13 +151,13 @@ def get_docs_from_parquet(parquet_urls, column, offset, limit):
     return df[column].tolist()
 
 
-# @spaces.GPU
+@spaces.GPU
 # TODO: Modify batch size to reduce memory consumption during embedding calculation, which value is better?
 def calculate_embeddings(docs):
     return sentence_model.encode(docs, show_progress_bar=True, batch_size=32)
 
 
-# @spaces.GPU
+@spaces.GPU
 def fit_model(docs, embeddings):
     global global_topic_model