ulisesbravo commited on
Commit
7cbfa47
·
verified ·
1 Parent(s): 6c6f65a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -2,16 +2,16 @@ import gradio as gr
2
  from huggingface_hub import InferenceClient
3
 
4
  # Inicialize o cliente com o modelo do Hugging Face
5
- client = InferenceClient("ulisesbravo/autotrain-nsuej-5ctie")
6
 
7
  def classify_text(text):
8
- # Realize a inferência chamando o método __call__
9
- response = client(text) # Chame o cliente como uma função
10
 
11
  # Inspecione a resposta para entender a estrutura
12
  print(response) # Para depuração
13
 
14
- # Extraia a classificação do response
15
  if isinstance(response, list) and len(response) > 0:
16
  predicted_class = response[0]['label'] if 'label' in response[0] else "Classificação não encontrada"
17
  else:
 
2
  from huggingface_hub import InferenceClient
3
 
4
  # Inicialize o cliente com o modelo do Hugging Face
5
+ client = InferenceClient(model="ulisesbravo/autotrain-nsuej-5ctie")
6
 
7
  def classify_text(text):
8
+ # Realize a inferência chamando o método post ou query
9
+ response = client.post(json={"inputs": text}) # Use o método correto para enviar o texto
10
 
11
  # Inspecione a resposta para entender a estrutura
12
  print(response) # Para depuração
13
 
14
+ # Extraia a classificação da resposta
15
  if isinstance(response, list) and len(response) > 0:
16
  predicted_class = response[0]['label'] if 'label' in response[0] else "Classificação não encontrada"
17
  else: