Spaces:
Runtime error
Runtime error
Commit
·
99b2813
1
Parent(s):
f3b8813
Update dashboard title and description
Browse files
app.py
CHANGED
|
@@ -135,7 +135,11 @@ def predict(phrase: str, finetuned_model: str):
|
|
| 135 |
demo = gr.Interface(
|
| 136 |
fn=predict,
|
| 137 |
inputs=[
|
| 138 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 139 |
gr.Dropdown([
|
| 140 |
"MatteoFasulo/twitter-roberta-base-hate_69",
|
| 141 |
"MatteoFasulo/twitter-roberta-base-hate_1337",
|
|
@@ -145,10 +149,16 @@ demo = gr.Interface(
|
|
| 145 |
"MatteoFasulo/xlm-roberta-base_42",
|
| 146 |
],
|
| 147 |
label="Model",
|
| 148 |
-
info="Choose the model to use for prediction.",
|
| 149 |
)
|
| 150 |
],
|
| 151 |
-
outputs=
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 152 |
)
|
| 153 |
|
| 154 |
demo.launch()
|
|
|
|
| 135 |
demo = gr.Interface(
|
| 136 |
fn=predict,
|
| 137 |
inputs=[
|
| 138 |
+
gr.Textbox(
|
| 139 |
+
label="Phrase",
|
| 140 |
+
placeholder="Enter a phrase to check if it is sexist or not.",
|
| 141 |
+
info="Enter a phrase to check if it is sexist or not.",
|
| 142 |
+
),
|
| 143 |
gr.Dropdown([
|
| 144 |
"MatteoFasulo/twitter-roberta-base-hate_69",
|
| 145 |
"MatteoFasulo/twitter-roberta-base-hate_1337",
|
|
|
|
| 149 |
"MatteoFasulo/xlm-roberta-base_42",
|
| 150 |
],
|
| 151 |
label="Model",
|
| 152 |
+
info="Choose the model to use for prediction. XLM-RoBERTa models are suitable for multilingual text.",
|
| 153 |
)
|
| 154 |
],
|
| 155 |
+
outputs=gr.Text(
|
| 156 |
+
label="Prediction",
|
| 157 |
+
info="The prediction of the model on the input phrase.",
|
| 158 |
+
),
|
| 159 |
+
title="Sexism Detection",
|
| 160 |
+
description="A small demo to check if a phrase is sexist or not using a fine-tuned RoBERTa model on hate speech detection.",
|
| 161 |
+
theme="huggingface",
|
| 162 |
)
|
| 163 |
|
| 164 |
demo.launch()
|