Gregoryjr commited on
Commit
3754612
1 Parent(s): d077642

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -2,7 +2,9 @@ import streamlit as st
2
  from transformers import pipeline
3
 
4
  st.title("Milestone #2 offensive statement prediction with pre-trained models")
5
- text = st.text_input("Enter a statement")
 
 
6
 
7
  options = ["zero-shot-classification", "cardiffnlp/twitter-roberta-base-offensive"]
8
  model = st.selectbox("Select a pre-trained model", options)
 
2
  from transformers import pipeline
3
 
4
  st.title("Milestone #2 offensive statement prediction with pre-trained models")
5
+ st.header("in the basic demo you can select a model to judge whether or not the text below is offensive")
6
+ text = "The mail man looks dumb"
7
+ st.write(text)
8
 
9
  options = ["zero-shot-classification", "cardiffnlp/twitter-roberta-base-offensive"]
10
  model = st.selectbox("Select a pre-trained model", options)