Spaces:

nehalelkaref
/

flat-arabic-entity-classification

Build error

nehalelkaref commited on Dec 13, 2023

Commit

abad191

1 Parent(s): ed916d4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,7 +21,7 @@ def predict_label(text):
     ensemble_span_scores = [score for scores in [span_scores, msa_span_scores] for score in scores]
     ensemble_pooled_scores = pool_span_scores(ensemble_span_scores, ip_len)
-    ent_scores = extract_ent_scores(entity_model,ip,ensemble_pooled_scores, pos_col=1, task_col=2)
     combined_sequences, ent_pred_tags = pool_ent_scores(ent_scores, ip_len)
     return combined_sequences

     ensemble_span_scores = [score for scores in [span_scores, msa_span_scores] for score in scores]
     ensemble_pooled_scores = pool_span_scores(ensemble_span_scores, ip_len)
+    ent_scores = extract_ent_scores(entity_model,ip,ensemble_pooled_scores)
     combined_sequences, ent_pred_tags = pool_ent_scores(ent_scores, ip_len)
     return combined_sequences