g0blas commited on
Commit
3abd188
1 Parent(s): f9c05dd

number of classes from 10 to 5

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,6 +21,6 @@ def classify_blog_tags(description):
21
  probs = torch.sigmoid(logits)[0]
22
  return dict(zip(tags, map(float, probs)))
23
 
24
- label = gr.outputs.Label(num_top_classes=10)
25
  iface = gr.Interface(fn=classify_blog_tags, inputs="text", outputs=label)
26
  iface.launch(inline=False)
 
21
  probs = torch.sigmoid(logits)[0]
22
  return dict(zip(tags, map(float, probs)))
23
 
24
+ label = gr.outputs.Label(num_top_classes=5)
25
  iface = gr.Interface(fn=classify_blog_tags, inputs="text", outputs=label)
26
  iface.launch(inline=False)