Update app.py

app.py

@@ -6,7 +6,7 @@ os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 
 # Load your pre-trained model
 def load_model():
-    model = tf.keras.models.load_model('depi-graduation-project.ipynb')  # Replace with your model's path
+    model = tf.keras.models.load_model('depi-graduation-project.h5')  # Replace with your model's path
     return model
 
 model = load_model()