Update app.py

app.py

@@ -5,15 +5,15 @@ from huggingface_hub import InferenceClient
 client = InferenceClient("ulisesbravo/autotrain-nsuej-5ctie")
 
 def classify_text(text):
-    # Realize a inferência chamando o método correto do cliente
-    response = client({"inputs": text})  # Chama o cliente como uma função
+    # Realize a inferência chamando o método __call__
+    response = client(text)  # Chame o cliente como uma função
 
     # Inspecione a resposta para entender a estrutura
     print(response)  # Para depuração
 
     # Extraia a classificação do response
-    if 'label' in response and 'score' in response:
-        predicted_class = response['label']
+    if isinstance(response, list) and len(response) > 0:
+        predicted_class = response[0]['label'] if 'label' in response[0] else "Classificação não encontrada"
     else:
         predicted_class = "Classificação não encontrada"