Update app.py

app.py

@@ -91,7 +91,7 @@ def generate_topics(dataset, config, split, column, nested_column):
     # Create instances of GPU-accelerated UMAP and HDBSCAN
     # umap_model = UMAP(n_components=5, n_neighbors=15, min_dist=0.0)
     # hdbscan_model = HDBSCAN(min_samples=10, gen_min_span_tree=True)
-    sentence_model = SentenceTransformer("all-MiniLM-L6-v2", device="cuda")
+    sentence_model = SentenceTransformer("all-MiniLM-L6-v2")
     while True:
         docs = get_docs_from_parquet(parquet_urls, column, offset, chunk_size)
         logging.info(f"------------> New chunk data {offset=} {chunk_size=}")