davidmasip commited on
Commit
dbc82f4
·
1 Parent(s): e680b78

improve code

Browse files
Files changed (1) hide show
  1. app.py +3 -6
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[0]["label"] == "LABEL_0":
13
- label = "Non-racist"
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(