Lisibonny commited on
Commit
6124424
1 Parent(s): 38cbe11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -142,8 +142,8 @@ def main():
142
  #answer_start_scores = tf.nn.softmax(outputs.start_logits)
143
  #answer_end_scores = tf.nn.softmax(outputs.end_logits)
144
  #######################
145
- start_probabilities = tf.nn.softmax(start_logits, dim=-1)[0]
146
- end_probabilities = tf.nn.softmax(end_logits, dim=-1)[0]
147
  scores = start_probabilities[:, None] * end_probabilities[None, :]
148
  scores = tf.experimental.numpy.triu(scores)
149
  max_index = scores.argmax().item()
 
142
  #answer_start_scores = tf.nn.softmax(outputs.start_logits)
143
  #answer_end_scores = tf.nn.softmax(outputs.end_logits)
144
  #######################
145
+ start_probabilities = tf.nn.softmax(outputs.start_logits, dim=-1)[0]
146
+ end_probabilities = tf.nn.softmax(outputs.end_logits, dim=-1)[0]
147
  scores = start_probabilities[:, None] * end_probabilities[None, :]
148
  scores = tf.experimental.numpy.triu(scores)
149
  max_index = scores.argmax().item()