MdRiad commited on
Commit
cb705d1
1 Parent(s): 916ddbf

Update app.py

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