Update app.py

app.py

@@ -6,11 +6,9 @@ client = InferenceClient("ulisesbravo/autotrain-nsuej-5ctie")
 
 def classify_text(text):
     # Faça a inferência no modelo de classificação
-    response = client(text)  # Aqui você pode precisar ajustar conforme a API do modelo
+    response = client.predict(text)  # Use o método predict para fazer a inferência
     # Extraia a classificação do response
-    # Isso depende de como o seu modelo retorna as classificações.
-    # Supondo que o response contenha uma lista de classes e a primeira posição seja a classe predita.
-    predicted_class = response['label']  # Altere isso conforme necessário
+    predicted_class = response['label']  # Ajuste isso conforme necessário
     return predicted_class
 
 # Interface Gradio