spelling
Browse files
app.py
CHANGED
@@ -44,10 +44,12 @@ else:
|
|
44 |
if st.button('Analyze'):
|
45 |
result = classifier(input)
|
46 |
output = None
|
47 |
-
if option == 'Fine-
|
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!')
|