Update README.md

README.md

@@ -39,19 +39,10 @@ Use the code below to get started with the model.
 ```python
 >>> from transformers import pipeline
 >>> classifier = pipeline("text-classification", model='hackathon-somos-nlp-2023/DiagTrast-Berto')
->>> text = ["Gasta más dinero de lo que tiene, a menudo, su falta de control hace que esté en deudas", 
-        "Le gusta estar solo y le molesta la gente a su alrededor, solo piensa en él",
-        "Tiene pocas habilidades sociales, ignora normas de convivencia", 
-        "Siempre que está en falta, culpa a los demás de sus problemas" ]
+>>> text = ["Tiene pocas habilidades sociales, ignora normas de convivencia"]
 >>> classifier.predict(text)
-[{'label': 'Trastornos de la personalidad antisocial',
-  'score': 0.9967895150184631},
- {'label': 'Trastornos de la personalidad esquizotípica',
-  'score': 0.9952175617218018},
- {'label': 'Trastornos de la personalidad antisocial',
-  'score': 0.9772088527679443},
- {'label': 'Trastornos de la personalidad antisocial',
-  'score': 0.855640172958374}]
+[ {'label': 'Trastornos de la personalidad antisocial',
+  'score': 0.9772088527679443}]
 ```
 
 ## Training Details