Gregoryjr commited on
Commit
d077642
·
unverified ·
1 Parent(s): 596ea97

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,11 +1,11 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
- st.title("Milestone #2 offensive statement prediction")
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 model", options)
9
 
10
  con = st.button("Submit")
11
  if con:
 
1
  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)
9
 
10
  con = st.button("Submit")
11
  if con: