Rob Caamano
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -40,7 +40,7 @@ if st.button("Submit", type="primary"):
|
|
40 |
|
41 |
tweet_portion = text[:50] + "..." if len(text) > 50 else text
|
42 |
|
43 |
-
if selected_model in ["Fine-tuned Toxicity Model", "
|
44 |
column_name = "Toxicity Class"
|
45 |
else:
|
46 |
column_name = "Prediction"
|
|
|
40 |
|
41 |
tweet_portion = text[:50] + "..." if len(text) > 50 else text
|
42 |
|
43 |
+
if selected_model in ["Fine-tuned Toxicity Model", "Model 3.0"]:
|
44 |
column_name = "Toxicity Class"
|
45 |
else:
|
46 |
column_name = "Prediction"
|