Update app.py

app.py

@@ -13,10 +13,6 @@ model_berto='hackathon-somos-nlp-2023/DiagTrast-Berto'
 tokenizer_berto = AutoTokenizer.from_pretrained(model_berto)
 classifier_berto = pipeline("text-classification", model=model_berto)
 
-model_xml='hackathon-somos-nlp-2023/DiagTrast-xlm-roberta-base'
-tokenizer_xml = AutoTokenizer.from_pretrained(model_xml)
-classifier_xml = pipeline("text-classification", model=model_xml)
-
 #####################
 
 st.title('🏥 Diagnóstico de Trastornos Mentales')
@@ -30,6 +26,12 @@ with DemoTab:
 
         submit_button = st.form_submit_button(label="Clasificar")
 
+    with st.form(key="examples"):
+        ex_1 = st.form_submit_button(label="Yo también")
+
+    if ex_1:
+        sintomas=ex_1
+
     if submit_button and not sintomas:
         st.warning("⚠️ Debe introducir los síntomas.")
         
@@ -39,7 +41,7 @@ with DemoTab:
             
             df = pd.DataFrame({
                 'Modelo': "DiagTrast-Berto",
-                'Diagnóstico': pred_berto[0]['label']]
+                'Diagnóstico': pred_berto[0]['label']
             })
 
         st.markdown("### Resultados:")