Update app.py

app.py

@@ -142,8 +142,8 @@ def main():
                #answer_start_scores = tf.nn.softmax(outputs.start_logits)
                #answer_end_scores = tf.nn.softmax(outputs.end_logits)
                #######################
-               start_probabilities = tf.nn.softmax(outputs.start_logits, dim=-1)[0]
-               end_probabilities = tf.nn.softmax(outputs.end_logits, dim=-1)[0]
+               start_probabilities = tf.nn.softmax(outputs.start_logits, axis=-1)[0]
+               end_probabilities = tf.nn.softmax(outputs.end_logits, axis=-1)[0]
                scores = start_probabilities[:, None] * end_probabilities[None, :]
                scores = tf.experimental.numpy.triu(scores)
                max_index = scores.argmax().item()