Hugging Face's logo

Spaces:
Jayesh13
/
Toxic_comment_classification
Runtime error

App Files Community
Jayesh13 commited on
Commit
1cf5bdb
·
1 Parent(s): 5628f9f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import streamlit as st
2
  import pickle
3
  from tensorflow.keras.models import load_model
4
 
5
- model = load_model('tox_model.pkl')
6
  text = st.text_area('Enter some text')
7
 
8
  if text:
 
2
  import pickle
3
  from tensorflow.keras.models import load_model
4
 
5
+ model = load_model('tox_model.h5')
6
  text = st.text_area('Enter some text')
7
 
8
  if text: