Armanul commited on
Commit
76165b8
·
verified ·
1 Parent(s): c2a239e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -22,7 +22,8 @@ def classify_dataset_type(description):
22
  probs = torch.sigmoid(logits)[0]
23
  return dict(zip(categories, map(float, probs)))
24
 
25
- label = gr.outputs.Label(num_top_classes=3)
26
- iface = gr.Interface(fn=classify_dataset_type, inputs="text", outputs=label)
27
- iface.launch(inline=False)
 
28
 
 
22
  probs = torch.sigmoid(logits)[0]
23
  return dict(zip(categories, map(float, probs)))
24
 
25
+ # label = gr.outputs.Label(num_top_classes=3)
26
+ # iface = gr.Interface(fn=classify_dataset_type, inputs="text", outputs=label)
27
+ iface = gr.Interface(fn=classify_dataset_type, gr.Text(), gr.Label())
28
+ iface.launch(inline=False, share=True)
29