Update app.py

app.py

@@ -6,9 +6,10 @@ client = InferenceClient("ulisesbravo/autotrain-nsuej-5ctie")
 
 def classify_text(text):
     # Faça a inferência no modelo de classificação
-    response = client.predict(text)  # Use o método predict para fazer a inferência
+    response = client(text)  # Tente chamar o cliente diretamente, pois pode ser que o __call__ funcione
     # Extraia a classificação do response
-    predicted_class = response['label']  # Ajuste isso conforme necessário
+    # Supondo que a resposta seja um dicionário com a chave 'label'
+    predicted_class = response.get('label')  # Use get para evitar KeyError
     return predicted_class
 
 # Interface Gradio