Pravincoder commited on
Commit
2273c9a
·
verified ·
1 Parent(s): 341ca84

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import numpy as np
4
  from tensorflow.keras.preprocessing.sequence import pad_sequences
5
 
6
  # Load the trained model
7
- model = tf.saved_model.load('./saved_model.pb')
8
 
9
  def spam_detection(message):
10
  # Preprocess the input message
 
4
  from tensorflow.keras.preprocessing.sequence import pad_sequences
5
 
6
  # Load the trained model
7
+ model = tf.keras.models.load_model('./model_file.pkl')
8
 
9
  def spam_detection(message):
10
  # Preprocess the input message