SatriyaTama commited on
Commit
26e4097
·
verified ·
1 Parent(s): c016824

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from tempfile import NamedTemporaryFile
7
  from tqdm import tqdm
8
 
9
  # Load model
10
- model = tf.keras.models.load_model('rain_detection_model(MobileNet).h5')
11
  IMG_SIZE = (256, 256)
12
 
13
  def batch_predict(frames):
 
7
  from tqdm import tqdm
8
 
9
  # Load model
10
+ model = tf.keras.models.load_model('rain_detection_model.h5')
11
  IMG_SIZE = (256, 256)
12
 
13
  def batch_predict(frames):