alpcansoydas commited on
Commit
28bb3db
·
verified ·
1 Parent(s): b0427a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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"Predicted Family Label: {predicted_label}"
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: