pszemraj commited on
Commit
9b8a77d
·
verified ·
1 Parent(s): f860236

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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