Rimi98 commited on
Commit
5b96f9c
·
1 Parent(s): ea201e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,6 +18,6 @@ def classify(review):
18
  probs = torch.sigmoid(logits)[0]
19
  return dict(zip(types,map(float,probs)))
20
 
21
- label = gr.outputs.Label()
22
  iface = gr.Interface(fn=classify,inputs='text',outputs = label)
23
  iface.launch(inline=False)
 
18
  probs = torch.sigmoid(logits)[0]
19
  return dict(zip(types,map(float,probs)))
20
 
21
+ label = gr.outputs.Label(num_top_classes=2)
22
  iface = gr.Interface(fn=classify,inputs='text',outputs = label)
23
  iface.launch(inline=False)