AMR-KELEG commited on
Commit
013e3f5
·
1 Parent(s): 73178f8

Skip the results of incomplete evaluations

Browse files
Files changed (2) hide show
  1. app.py +3 -0
  2. background_inference.py +3 -1
app.py CHANGED
@@ -53,6 +53,9 @@ with tab1:
53
  precision_scores = {}
54
  predictions = row["predictions"]
55
 
 
 
 
56
  for dialect in DIALECTS_WITH_LABELS:
57
  y_true = labels[dialect]
58
  y_pred = [dialect in prediction for prediction in predictions]
 
53
  precision_scores = {}
54
  predictions = row["predictions"]
55
 
56
+ if not row["status"] == "completed":
57
+ continue
58
+
59
  for dialect in DIALECTS_WITH_LABELS:
60
  y_true = labels[dialect]
61
  y_pred = [dialect in prediction for prediction in predictions]
background_inference.py CHANGED
@@ -20,7 +20,9 @@ utils.update_model_queue(
20
 
21
  try:
22
  tokenizer = AutoTokenizer.from_pretrained(model_name, revision=commit_id)
23
- model = AutoModelForSequenceClassification.from_pretrained(model_name, revision=commit_id)
 
 
24
 
25
  # Load the dataset
26
  dataset_name = os.environ["DATASET_NAME"]
 
20
 
21
  try:
22
  tokenizer = AutoTokenizer.from_pretrained(model_name, revision=commit_id)
23
+ model = AutoModelForSequenceClassification.from_pretrained(
24
+ model_name, revision=commit_id
25
+ )
26
 
27
  # Load the dataset
28
  dataset_name = os.environ["DATASET_NAME"]