RohithMidigudla commited on
Commit
0c1de34
·
verified ·
1 Parent(s): 5b3e824

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ vecterizor.adapt(X.values)
18
  model_path = 'commenttoxicity (1).h5'
19
 
20
  # Load the model with custom_objects
21
- model = load_model(model_path, custom_objects={'LSTM': LSTM})
22
 
23
  def score_comment(comment):
24
  vectorized_comment = vecterizor([comment])
 
18
  model_path = 'commenttoxicity (1).h5'
19
 
20
  # Load the model with custom_objects
21
+ model = load_model(model_path)
22
 
23
  def score_comment(comment):
24
  vectorized_comment = vecterizor([comment])