Update app.py
Browse files
app.py
CHANGED
@@ -47,7 +47,7 @@ def main():
|
|
47 |
text='Manuel Romero está colaborando activamente con huggingface/transformers para traer el poder de las últimas técnicas de procesamiento de lenguaje natural al idioma español'
|
48 |
inputs = tokenizer(query, text)
|
49 |
st.info(inputs)
|
50 |
-
outputs = qa_model(
|
51 |
answer_start_index = int(tf.math.argmax(outputs.start_logits, axis=-1)[0])
|
52 |
answer_end_index = int(tf.math.argmax(outputs.end_logits, axis=-1)[0])
|
53 |
predict_answer_tokens = inputs.input_ids[0, answer_start_index : answer_end_index + 1]
|
|
|
47 |
text='Manuel Romero está colaborando activamente con huggingface/transformers para traer el poder de las últimas técnicas de procesamiento de lenguaje natural al idioma español'
|
48 |
inputs = tokenizer(query, text)
|
49 |
st.info(inputs)
|
50 |
+
outputs = qa_model(input_ids=inputs['input_ids'])
|
51 |
answer_start_index = int(tf.math.argmax(outputs.start_logits, axis=-1)[0])
|
52 |
answer_end_index = int(tf.math.argmax(outputs.end_logits, axis=-1)[0])
|
53 |
predict_answer_tokens = inputs.input_ids[0, answer_start_index : answer_end_index + 1]
|