jbraha commited on
Commit
636ed81
1 Parent(s): ea78898
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -44,10 +44,12 @@ else:
44
  if st.button('Analyze'):
45
  result = classifier(input)
46
  output = None
47
- if option == 'Fine-tuned':
48
  output = {'Toxic': result['LABEL_0']}
49
  del result['LABEL_0']
50
  output[max(result, key=result.get)] = result[max(result, key=result.get)]
 
 
51
  st.write(output)
52
  else:
53
  st.write('Excited to analyze!')
 
44
  if st.button('Analyze'):
45
  result = classifier(input)
46
  output = None
47
+ if option == 'Fine-Tuned':
48
  output = {'Toxic': result['LABEL_0']}
49
  del result['LABEL_0']
50
  output[max(result, key=result.get)] = result[max(result, key=result.get)]
51
+ else:
52
+ output = result
53
  st.write(output)
54
  else:
55
  st.write('Excited to analyze!')