Spaces:

Lisibonny
/

buscador-periodicos-dominicanos

Running

Lisibonny commited on May 5

Commit

24e574d

•

1 Parent(s): 46148a4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -51,8 +51,9 @@ def main():
            answer_start_index = int(tf.math.argmax(outputs.start_logits, axis=-1)[0])
            answer_end_index = int(tf.math.argmax(outputs.end_logits, axis=-1)[0])
            predict_answer_tokens = inputs.input_ids[0, answer_start_index : answer_end_index + 1]
-           st.info(predict_answer_tokens)
         else:

            answer_start_index = int(tf.math.argmax(outputs.start_logits, axis=-1)[0])
            answer_end_index = int(tf.math.argmax(outputs.end_logits, axis=-1)[0])
            predict_answer_tokens = inputs.input_ids[0, answer_start_index : answer_end_index + 1]
+           answer=tokenizer.decode(predict_answer_tokens)
+           st.info(answer)
         else: