Update app.py

app.py

@@ -44,7 +44,6 @@ def crear_indice():
 def load_qa_model():
 
     tokenizer = AutoTokenizer.from_pretrained('dccuchile/bert-base-spanish-wwm-cased', use_fast="false")
-    #model = AutoModelForQuestionAnswering.from_pretrained("Lisibonny/modelo_qa_beto_squad")
     model = TFAutoModelForQuestionAnswering.from_pretrained("Lisibonny/modelo_qa_beto_squad")
     return tokenizer, model
 
@@ -129,6 +128,7 @@ def main():
                text=df.loc[i, "resumen"]
                inputs =  tokenizer(query, text, return_tensors='tf')
                outputs = qa_model(input_ids=inputs['input_ids'], attention_mask=inputs['attention_mask'])
+               st.markdown(outputs)
                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]