elshehawy commited on
Commit
1033026
Β·
1 Parent(s): 57aa9a4

add tqdm to gpt

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,7 +10,7 @@ import json
10
  # from simcse import SimCSE # use for gpt
11
  from evaluate_data import store_sample_data, get_metrics_trf
12
  from sentence_transformers import SentenceTransformer
13
-
14
  # store_sample_data()
15
 
16
 
@@ -100,7 +100,7 @@ def find_orgs(uploaded_file):
100
 
101
  gpt_orgs, true_orgs = [], []
102
 
103
- for sent in sample_data:
104
  gpt_orgs.append(find_orgs_gpt(sent['text']))
105
  true_orgs.append(sent['orgs'])
106
 
 
10
  # from simcse import SimCSE # use for gpt
11
  from evaluate_data import store_sample_data, get_metrics_trf
12
  from sentence_transformers import SentenceTransformer
13
+ from tqdm import tqdm
14
  # store_sample_data()
15
 
16
 
 
100
 
101
  gpt_orgs, true_orgs = [], []
102
 
103
+ for sent in tqdm(sample_data):
104
  gpt_orgs.append(find_orgs_gpt(sent['text']))
105
  true_orgs.append(sent['orgs'])
106