Update predict.py

predict.py

@@ -6,12 +6,12 @@ from tensorflow.keras.optimizers import Adam
 
 from constants import LEARNING_RATE
 
+model = get_model()
 
 def predict_label(path):
     frames = load_video(path)
-    model = get_model()
-    prediction = model.predict(tf.expand_dims(example, axis=0))[0]
-    label = np.argmax(pred, axis=0)
+    prediction = model.predict(tf.expand_dims(frames, axis=0))[0]
+    label = np.argmax(prediction, axis=0)
 
     return label