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!')
|