Spaces:
Runtime error
Runtime error
Commit
·
dbc82f4
1
Parent(s):
e680b78
improve code
Browse files
app.py
CHANGED
@@ -8,12 +8,9 @@ racism_analysis_pipe = pipeline("text-classification",
|
|
8 |
|
9 |
|
10 |
def racism_analysis(text):
|
11 |
-
results = racism_analysis_pipe(text)
|
12 |
-
if results[
|
13 |
-
|
14 |
-
else:
|
15 |
-
label = "Racist"
|
16 |
-
return label, round(results[0]["score"], 5)
|
17 |
|
18 |
|
19 |
gradio_ui = gr.Interface(
|
|
|
8 |
|
9 |
|
10 |
def racism_analysis(text):
|
11 |
+
results = racism_analysis_pipe(text)[0]
|
12 |
+
label = "Non-racist" if results["label"] == "LABEL_0" else "Racist"
|
13 |
+
return label, round(results["score"], 5)
|
|
|
|
|
|
|
14 |
|
15 |
|
16 |
gradio_ui = gr.Interface(
|