Update app.py

app.py

@@ -6,7 +6,7 @@ from bertopic import BERTopic
 import pandas as pd
 import gradio as gr
 from bertopic.representation import KeyBERTInspired
-
+import spaces
 
 logging.basicConfig(
     level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
@@ -34,7 +34,7 @@ def get_docs_from_parquet(parquet_urls, column, offset, limit):
     logging.debug(f"Dataframe: {df.head(5)}")
     return df[column].tolist()
 
-
+@spaces.GPU
 def generate_topics(dataset, config, split, column, nested_column):
     logging.info(
         f"Generating topics for {dataset} with config {config} {split} {column} {nested_column}"