Update app.py

app.py

@@ -1,12 +1,11 @@
 import gradio as gr
 from transformers import pipeline
 
-# Carregar um modelo de classificação de texto pré-treinado
-classifier = pipeline("text-classification", model="distilbert-base-uncased")
+# Carregar um modelo em português
+classifier = pipeline("text-classification", model="neuralmind/bert-base-portuguese-cased")
 
 # Função para classificar a criticidade da compra
 def classificar_criticidade(descricao):
-    # Aqui você pode adicionar lógica personalizada para mapear a saída do modelo para "Baixa", "Média", "Alta"
     resultado = classifier(descricao)[0]
     label = resultado['label']
     score = resultado['score']