Update app.py
Browse files
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()
|