elshehawy commited on
Commit
10eaeda
Β·
1 Parent(s): 970df24

fix issues

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. evaluate_data.py +2 -1
app.py CHANGED
@@ -83,6 +83,7 @@ all_metrics = {}
83
 
84
  # sim_model = SimCSE('sentence-transformers/all-MiniLM-L6-v2')
85
  # all_metrics['gpt'] = calc_metrics(true_orgs, predicted_orgs_gpt, sim_model)
 
86
  all_metrics['trf'] = get_metrics_trf()
87
 
88
 
 
83
 
84
  # sim_model = SimCSE('sentence-transformers/all-MiniLM-L6-v2')
85
  # all_metrics['gpt'] = calc_metrics(true_orgs, predicted_orgs_gpt, sim_model)
86
+ print('Finiding all metrics trf')
87
  all_metrics['trf'] = get_metrics_trf()
88
 
89
 
evaluate_data.py CHANGED
@@ -9,7 +9,8 @@ from utils import tokenize_and_align_labels
9
  from rich import print
10
  import huggingface_hub
11
  import torch
12
-
 
13
 
14
  # _ = load_dotenv(find_dotenv()) # read local .env file
15
  hf_token= os.environ['HF_TOKEN']
 
9
  from rich import print
10
  import huggingface_hub
11
  import torch
12
+ import json
13
+ from tqdm import tqdm
14
 
15
  # _ = load_dotenv(find_dotenv()) # read local .env file
16
  hf_token= os.environ['HF_TOKEN']