Update app.py

app.py

@@ -14,7 +14,8 @@ pipe = pipeline(
 )
 
 # Define custom labels for classification
-pipe.model.config.id2label = {0: 'greeting', 1: 'farewell', 2: 'other'}
+pipe.model.config.id2label = {0: 'saudação', 1: 'fim de conversa', 2: 'outro'}
+pipe.model.config.label2id = {'saudação': 0, 'fim de conversa': 1, 'outro': 2}
 
 # Function to classify input text
 def classify_text(text):
@@ -24,10 +25,10 @@ def classify_text(text):
 # Create Gradio interface
 iface = gr.Interface(
     fn=classify_text,  # Function to be called
-    inputs=gr.Textbox(label="Enter Text"),  # Textbox input for user
-    outputs=gr.Label(label="Classification"),  # Output label showing classification
-    title="Text Classifier",  # Title of the app
-    description="Classify your text as 'greeting', 'farewell', or 'other'."  # Description of the app
+    inputs=gr.Textbox(label="Digite o texto"),  # Textbox input for user
+    outputs=gr.Label(label="Classificação"),  # Output label showing classification
+    title="Classificação Textual",  # Title of the app
+    description="Classifique seu texto como 'saudação', 'fim de conversa', ou 'outro'."  # Description of the app
 )
 
 # Launch the Gradio app