Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)
|
|
16 |
def classify_product_family(text):
|
17 |
results = classifier(text)
|
18 |
predicted_label = results[0]['label']
|
19 |
-
return f"
|
20 |
|
21 |
# Gradio interface
|
22 |
with gr.Blocks() as demo:
|
|
|
16 |
def classify_product_family(text):
|
17 |
results = classifier(text)
|
18 |
predicted_label = results[0]['label']
|
19 |
+
return f"{predicted_label}"
|
20 |
|
21 |
# Gradio interface
|
22 |
with gr.Blocks() as demo:
|