Update app.py

app.py

@@ -137,7 +137,6 @@ def main():
                text=remove_URL(text)
                inputs =  tokenizer(query, text[0:512], return_tensors='tf')
                outputs = qa_model(input_ids=inputs['input_ids'], attention_mask=inputs['attention_mask'])
-               st.write("HI")
                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])
                #answer_start_scores = tf.nn.softmax(outputs.start_logits)
@@ -153,6 +152,7 @@ def main():
                    #score = answer_start_scores*answer_end_scores
                    #st.write(f'Aqui {answer_start_scores}' )
                    #st.write(answer_end_scores)
+                   st.write(outputs)
                    cantidad_respuestas = cantidad_respuestas + 1
                    df_answer.loc[i, "answer"] = answer
                    lista_noticias_respuestas.append(df_answer.loc[i].to_frame().T)