Lisibonny commited on
Commit
73b59b1
1 Parent(s): 1b76811

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(**inputs)
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]