Update app.py
Browse files
app.py
CHANGED
@@ -141,6 +141,16 @@ def main():
|
|
141 |
answer_end_index = int(tf.math.argmax(outputs.end_logits, axis=-1)[0])
|
142 |
#answer_start_scores = tf.nn.softmax(outputs.start_logits)
|
143 |
#answer_end_scores = tf.nn.softmax(outputs.end_logits)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
144 |
predict_answer_tokens = inputs.input_ids[0, answer_start_index : answer_end_index + 1]
|
145 |
answer=tokenizer.decode(predict_answer_tokens)
|
146 |
|
@@ -152,7 +162,7 @@ def main():
|
|
152 |
#score = answer_start_scores*answer_end_scores
|
153 |
#st.write(f'Aqui {answer_start_scores}' )
|
154 |
#st.write(answer_end_scores)
|
155 |
-
|
156 |
cantidad_respuestas = cantidad_respuestas + 1
|
157 |
df_answer.loc[i, "answer"] = answer
|
158 |
lista_noticias_respuestas.append(df_answer.loc[i].to_frame().T)
|
|
|
141 |
answer_end_index = int(tf.math.argmax(outputs.end_logits, axis=-1)[0])
|
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()
|
150 |
+
start_index = max_index // scores.shape[1]
|
151 |
+
end_index = max_index % scores.shape[1]
|
152 |
+
st.write(scores[start_index, end_index])
|
153 |
+
#######################
|
154 |
predict_answer_tokens = inputs.input_ids[0, answer_start_index : answer_end_index + 1]
|
155 |
answer=tokenizer.decode(predict_answer_tokens)
|
156 |
|
|
|
162 |
#score = answer_start_scores*answer_end_scores
|
163 |
#st.write(f'Aqui {answer_start_scores}' )
|
164 |
#st.write(answer_end_scores)
|
165 |
+
|
166 |
cantidad_respuestas = cantidad_respuestas + 1
|
167 |
df_answer.loc[i, "answer"] = answer
|
168 |
lista_noticias_respuestas.append(df_answer.loc[i].to_frame().T)
|