CIAZIZ commited on
Commit
1576143
·
verified ·
1 Parent(s): e21466b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -6,8 +6,10 @@ pipe = pipeline("text-classification", model="CAMeL-Lab/bert-base-arabic-camelbe
6
 
7
  def analyze(message):
8
  result = pipe(message)
9
- result[2] = result[2] * 100
10
- return result
 
 
11
 
12
  start = gr.Interface(fn=analyze, inputs="text", outputs="text")
13
  start.launch(debug=True)
 
6
 
7
  def analyze(message):
8
  result = pipe(message)
9
+ result = result[0]
10
+ label = result['label']
11
+ score = result['score'] * 100
12
+ return f"{label}, (score:.2f)"
13
 
14
  start = gr.Interface(fn=analyze, inputs="text", outputs="text")
15
  start.launch(debug=True)