Update app.py

app.py

@@ -41,8 +41,8 @@ def crear_indice():
 
 def load_qa_model():
 
-    tokenizer = AutoTokenizer.from_pretrained('mrm8488/distill-bert-base-spanish-wwm-cased-finetuned-spa-squad2-es', use_fast="false")
-    model = TFDistilBertForQuestionAnswering.from_pretrained("mrm8488/distill-bert-base-spanish-wwm-cased-finetuned-spa-squad2-es", from_pt=True)
+    tokenizer = AutoTokenizer.from_pretrained('dccuchile/bert-base-spanish-wwm-cased', use_fast="false")
+    model = AutoModelForQuestionAnswering.from_pretrained("Lisibonny/modelo_qa_beto_squad", from_pt=True)
     return tokenizer, model
 
 # 4. Use streamlit to create a web app
@@ -118,7 +118,11 @@ def main():
 
         if ('?' in query):
            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.' 
+           text="""
+           El director de la Corporación de Acueductos y Alcantarillado de Santiago (Coraasan), Andrés Burgos, informó ayer que un equipo realizó un levantamiento 
+           de los daños causados por la explosión de una tubería de agua en el sector Bella Vista, para asumir totalmente los gastos de rehabilitación y amueblamiento 
+           de la vivienda afectada.
+           """
            inputs =  tokenizer(query, text, return_tensors='tf')
            outputs = qa_model(input_ids=inputs['input_ids'], attention_mask=inputs['attention_mask'])
            answer_start_index = int(tf.math.argmax(outputs.start_logits, axis=-1)[0])