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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ def main():
45
  if ('?' in query):
46
  st.write("Contestando a: ", query)
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])
 
45
  if ('?' in query):
46
  st.write("Contestando a: ", query)
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, return_tensors='tf')
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])