Update app.py

app.py

@@ -1,6 +1,7 @@
 import cv2
 import numpy as np
-import onnxruntime as ort  # Import onnxruntime for model inference
+import os
+import onnxruntime as ort
 import streamlit as st
 from PIL import Image
 
@@ -38,7 +39,9 @@ def predict_emotion_with_smile(image_input, face_landmarks=None):
 
 # Load ONNX model
 def load_model():
-    model_path = '"onnx_model.onnx"'  # Make sure to set the correct path
+    model_path = 'onnx_model.onnx'  # Make sure this is the correct path
+    if not os.path.exists(model_path):
+        raise FileNotFoundError(f"Model file {model_path} not found!")
     emotion_model = ort.InferenceSession(model_path)
     return emotion_model