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