Update app.py

app.py

@@ -6,9 +6,12 @@ client = InferenceClient("ulisesbravo/autotrain-nsuej-5ctie")
 
 def classify_text(text):
     # Faça a inferência no modelo de classificação
-    response = client.predict(text=text)  # Use o método predict com o parâmetro correto
+    response = client(text)  # Utilize __call__ se o método predict não existir
+    # Imprima a resposta para verificar sua estrutura
+    print(response)  # Inspecione a resposta para entender a estrutura
     # Extraia a classificação do response
-    predicted_class = response['label']  # Ajuste conforme a estrutura da resposta do modelo
+    # Ajuste isso conforme a estrutura do response
+    predicted_class = response['label'] if 'label' in response else "Classificação não encontrada"
     return predicted_class
 
 # Interface Gradio