g0blas commited on
Commit
1d73f41
1 Parent(s): 7e61691

Update app.py

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