minhaj-ripon commited on
Commit
21c2780
·
1 Parent(s): 4c5b0ed

app.py updated

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