Update app.py

app.py

@@ -48,6 +48,8 @@ def get_entities(example):
     return output
 
 def clasifica_sistema_universal(example):
+    from transformers import AutoTokenizer, AutoModelForSequenceClassification
+
     tokenizer = AutoTokenizer.from_pretrained("hackathon-pln-es/jurisbert-class-tratados-internacionales-sistema-universal")
 
     model = AutoModelForSequenceClassification.from_pretrained("hackathon-pln-es/jurisbert-class-tratados-internacionales-sistema-universal")
@@ -83,6 +85,6 @@ def process(example):
     return entidades,class_sistema_universal, class_conv_americana
 
 
-iface = gr.Interface(fn=process, inputs="text", outputs=["highlight","label","label"], examples=examples, title="Modelo Jurídico Mexicano")
+iface = gr.Interface(fn=process, inputs=gr.inputs.Textbox(lines=10, placeholder="Proporcione el texto a analizar:"), outputs=["highlight","label","label"], examples=examples, title="Modelo Jurídico Mexicano")
 
 iface.launch(debug=True)