Armanul commited on
Commit
1e6b6c9
1 Parent(s): da60655

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -22,8 +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 = gr.Interface(fn=classify_dataset_type, inputs="textbox", outputs=gr.Label())
28
  iface.launch(inline=False, share=True)
29
 
 
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, inputs="textbox", outputs=gr.Label())
28
  iface.launch(inline=False, share=True)
29