Rob Caamano commited on
Commit
a3bd575
·
unverified ·
1 Parent(s): 8670e73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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", "Fine-tuned Toxicity Model RObert2"]:
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"