Rishi-19 commited on
Commit
1474905
·
1 Parent(s): 0acb963

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ model = torch.load('saved_model.pth', map_location=torch.device('cpu'))
9
  with open('tokenizer.pkl', 'rb') as f:
10
  tokenizer = pickle.load(f)
11
 
12
- st.title("Text Classification Streamlit App")
13
 
14
  input_text = st.text_input("Enter text:")
15
  if st.button("Predict"):
 
9
  with open('tokenizer.pkl', 'rb') as f:
10
  tokenizer = pickle.load(f)
11
 
12
+ st.title("GPT2 Profanity Classifier")
13
 
14
  input_text = st.text_input("Enter text:")
15
  if st.button("Predict"):