Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import json
+import re
 from huggingface_hub import InferenceClient
 
 client = InferenceClient(
@@ -131,7 +132,8 @@ def create_interface():
     def fn(text, rag, prompt_template):
         
         output = process_input(text, rag, prompt_template)
-        classification_types = [item['clasificacion_tipo'] for item in output]
+        matches = re.findall(r"'tipo': '([^']+)'", output)
+        classification_types = matches
         classification_types_options = [(option, option) for option in classification_types]
         classification_types_checkboxes = gr.CheckboxGroup(label="Clasificacion Tipo", choices=classification_types_options, interactive = True)