nehalelkaref commited on
Commit
383c547
1 Parent(s): 186dc98

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ def predict_label(text):
12
  span_scores = extract_spannet_scores(span_model,ip,ip_len, pos_col=1, task_col=2)
13
  span_pooled_scores = pool_span_scores(span_scores, ip_len)
14
 
15
- msa_span_scores = extract_spannet_scores(msa_span_model,ip,ip_len, pos_col=1, task_col=2)
16
  msa_pooled_scores = pool_span_scores(msa_span_scores, ip_len)
17
 
18
  ensemble_span_scores = [score for scores in [span_scores, msa_span_scores] for score in scores]
 
12
  span_scores = extract_spannet_scores(span_model,ip,ip_len, pos_col=1, task_col=2)
13
  span_pooled_scores = pool_span_scores(span_scores, ip_len)
14
 
15
+ msa_span_scores = extract_spannet_scores(msa_span_model,ip,ip_len, pos_col=1, task_col=2, pos='not none')
16
  msa_pooled_scores = pool_span_scores(msa_span_scores, ip_len)
17
 
18
  ensemble_span_scores = [score for scores in [span_scores, msa_span_scores] for score in scores]