Spaces:

nehalelkaref
/

flat-arabic-entity-classification

Build error

nehalelkaref commited on Nov 22, 2023

Commit

60dcb6e

1 Parent(s): 77f123e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ def predict_label(text):
     span_scores = extract_spannet_scores(span_model,ip,ip_len, pos_col=1, task_col=2)
     pooled_scores = pool_span_scores(span_scores, ip_len)
-    ent_scores = extract_ent_scores(ent_model, ip,pooled_scores, pos_col=1, task_col=2)
     combined_sequences, ent_pred_tags = pool_ent_scores(ent_scores, ip_len)
     return combined_sequences

     span_scores = extract_spannet_scores(span_model,ip,ip_len, pos_col=1, task_col=2)
     pooled_scores = pool_span_scores(span_scores, ip_len)
+    ent_scores = extract_ent_scores(entity_model,ip,pooled_scores, pos_col=1, task_col=2)
     combined_sequences, ent_pred_tags = pool_ent_scores(ent_scores, ip_len)
     return combined_sequences