Update app.py

app.py

@@ -33,7 +33,7 @@ def main():
     df=cargar_articulos()
     articulos_indexados = pd.read_csv('articulos_indexados.csv')
     articulos_indexados = articulos_indexados.set_index('Unnamed: 0')
-    qa = load_qa_model()
+    qa_model = load_qa_model()
     
 
     query = st.text_input(
@@ -44,13 +44,15 @@ def main():
 
         if ('?' in query):
            st.write("Contestando a: ", query)
-
-           answers = qa({ 'question': query,
-                          'context': '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'
-    }
-) 
-           st.info(answers['answer'])     
+           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' 
+           inputs =  tokenizer(query, text, return_tensors="tf")
+           outputs = qa_model(**inputs)
+           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] 
+            
+           st.info(predict_answer_tokens)     
 
         else: