Update app.py

app.py

@@ -6,10 +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)  # Tente chamar o cliente diretamente, pois pode ser que o __call__ funcione
+    response = client.predict(text=text)  # Use o método predict com o parâmetro correto
     # Extraia a classificação do response
-    # Supondo que a resposta seja um dicionário com a chave 'label'
-    predicted_class = response.get('label')  # Use get para evitar KeyError
+    predicted_class = response['label']  # Ajuste conforme a estrutura da resposta do modelo
     return predicted_class
 
 # Interface Gradio