Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
9 |
-
response = client(text) #
|
10 |
|
11 |
# Inspecione a resposta para entender a estrutura
|
12 |
print(response) # Para depuração
|
13 |
|
14 |
-
# Extraia a classificação
|
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:
|