Update app.py

app.py

@@ -2,6 +2,12 @@ import gradio as gr
 import tensorflow as tf
 import numpy as np
 import cv2
+import os
+
+if not os.path.exists("keras_model.h5"):
+    raise FileNotFoundError("Model file not found!")
+
+model = tf.keras.models.load_model("keras_model.h5")
 
 # Load Teachable Machine model
 model = tf.keras.models.load_model("./keras_model.h5")