DrGabrielLopez commited on
Commit
66b46d9
·
1 Parent(s): a9f525a
Files changed (2) hide show
  1. .DS_Store +0 -0
  2. app.py +5 -5
.DS_Store ADDED
Binary file (6.15 kB). View file
 
app.py CHANGED
@@ -26,9 +26,9 @@ ARTICLE = r"""<center>
26
 
27
 
28
  # load data
29
- def load_data(path):
30
- """Load CSV dataset"""
31
- data = pd.read_csv(path, error_bad_lines=False)
32
  assert "text" in data.columns, "The data must have a column named 'text'"
33
  return data
34
 
@@ -148,7 +148,7 @@ with blocks:
148
  gr.Markdown(ARTICLE)
149
  # event listeners
150
  in_file = in_file.change(inputs=in_file, outputs=in_data, fn=load_data)
151
- submit_button.click(inputs=in_data, outputs=out_dataset, fn=run_bert_tokenization)
152
- out_dataset.change(inputs=out_dataset, outputs=embedding_plot, fn=run_bertopic)
153
 
154
  blocks.launch()
 
26
 
27
 
28
  # load data
29
+ def load_data(file_obj):
30
+ """Load dataset (keep only 500 rows for efficiency)"""
31
+ data = pd.read_csv(file_obj, error_bad_lines=False, nrows=500)
32
  assert "text" in data.columns, "The data must have a column named 'text'"
33
  return data
34
 
 
148
  gr.Markdown(ARTICLE)
149
  # event listeners
150
  in_file = in_file.change(inputs=in_file, outputs=in_data, fn=load_data)
151
+ # submit_button.click(inputs=in_data, outputs=out_dataset, fn=run_bert_tokenization)
152
+ # out_dataset.change(inputs=out_dataset, outputs=embedding_plot, fn=run_bertopic)
153
 
154
  blocks.launch()