Update app.py

app.py

@@ -30,16 +30,21 @@ if len(pregs)<3:
 
 boton_preg = st.button('Enviar preguntas')
 
-if boton_preg:
-
-  st.write('Procesando texto ', texto[0:50], '...')
+modelo_hf2 = 'mrm8488/distill-bert-base-spanish-wwm-cased-finetuned-spa-squad2-es'
+model2 = AutoModelForQuestionAnswering.from_pretrained(modelo_hf2)
+#modelo_qa = AutoModelForQuestionAnswering.from_pretrained("mrm8488/bert-base-spanish-wwm-cased-finetuned-spa-squad2-es")
 
-  tokenizer = AutoTokenizer.from_pretrained("mrm8488/bert-base-spanish-wwm-cased-finetuned-spa-squad2-es",
-                                           #{"use_fast": False}
+tokenizer = AutoTokenizer.from_pretrained("mrm8488/bert-base-spanish-wwm-cased-finetuned-spa-squad2-es",
+                                            #{"use_fast": False}
                                            )
-  modelo_qa = AutoModelForQuestionAnswering.from_pretrained("mrm8488/bert-base-spanish-wwm-cased-finetuned-spa-squad2-es")
-  preguntas = pipeline('question-answering', model=modelo_qa, tokenizer=tokenizer)
+tokenizer2 = AutoTokenizer.from_pretrained(modelo_hf2,
+                                             #{"use_fast": False}
+                                            )
 
+if boton_preg:
+
+  st.write('Procesando texto ', texto[0:50], '...')
+  preguntas = pipeline('question-answering', model=modelo_qa, tokenizer=tokenizer2)
   for preg in pregs:
     respuesta = preguntas({'question': preg, 'context': texto})
     st.write(preg)