Update app.py
Browse files
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 |
-
|
26 |
-
|
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 |
|