Update app.py

app.py

@@ -47,8 +47,8 @@ def main():
            st.write("Contestando a: ", query)
            text='Un texto es una composición de signos codificados en un sistema de escritura que forma una unidad de sentido.' 
            inputs =  tokenizer(query, text, return_tensors='tf')
-           #outputs = qa_model(input_ids=inputs['input_ids'], attention_mask=inputs['attention_mask'])
-           outputs = qa_model(input_ids=inputs['input_ids']) 
+           outputs = qa_model(input_ids=inputs['input_ids'], attention_mask=inputs['attention_mask'])
+           #outputs = qa_model(input_ids=inputs['input_ids']) 
            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]