Update app.py
Browse files
app.py
CHANGED
@@ -62,7 +62,7 @@ def vectorize_dataset(hub_repo_id: str, split: str, column: str):
|
|
62 |
gr.Info("Vectorizing dataset...")
|
63 |
ds = load_dataset(hub_repo_id)
|
64 |
df = ds[split].to_polars()
|
65 |
-
embeddings = model.encode(df[column].cast(str), show_progress_bar=True, batch_size=128)
|
66 |
return embeddings
|
67 |
|
68 |
|
|
|
62 |
gr.Info("Vectorizing dataset...")
|
63 |
ds = load_dataset(hub_repo_id)
|
64 |
df = ds[split].to_polars()
|
65 |
+
embeddings = model.encode(df[column].cast(str).to_list(), show_progress_bar=True, batch_size=128)
|
66 |
return embeddings
|
67 |
|
68 |
|