gradio fixes

README.md

@@ -13,3 +13,16 @@ license: mit
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 =======
 # antisomnus - Driver Drowsiness Detection
+
+
+## Instructions
+
+1. Navigate to project directory
+2. Run the following commands after deactivating your current Python virtual environment if you are already on one.
+
+```bash
+python3 -m venv .venv
+source .venv/bin/activate
+pip install -r requirements.txt
+gradio app.py
+```

app.py

@@ -11,12 +11,11 @@ import tensorflow as tf
 import cv2
 
 # Add the path to the model directory
-path = Path(os.getcwd())
-sys.path.insert(0,str(path.absolute())+"/model/data")
+sys.path.append("model/data")
 from mp_process import process_mp_img
 
 
-model = tf.keras.models.load_model(str(path.absolute())+"/model/training/saved_models/en_model_v0.h5")
+model = tf.keras.models.load_model("model/training/saved_models/en_model_v0.h5")
 
 def preprocess_frame(frame):
     """
@@ -47,14 +46,14 @@ def detect_drowsiness(frame):
 
 
 # Define the input component as an Image component
-input_image = gr.inputs.Image(shape=(480, 640), source="webcam", label="live feed")
+input_image = gr.Image(shape=(480, 640), source="webcam", label="live feed")
 
 # Define the output components as an Image and a Label component
-output_image = gr.components.Image(label="Drowsiness Detection")
-output_model = gr.components.Label(label="Drowsiness Status - en_model_v0.h5")
-output_mp = gr.components.Label(label="Drowsiness Status - MediaPipe")
-output_eye = gr.components.Textbox(label="Eye Aspect Ratio")
-output_mouth = gr.components.Textbox(label="Mouth Aspect Ratio")
+output_image = gr.Image(shape=(480,640),label="Drowsiness Detection")
+output_model = gr.Label(label="Drowsiness Status - en_model_v0.h5")
+output_mp = gr.Label(label="Drowsiness Status - MediaPipe")
+output_eye = gr.Textbox(label="Eye Aspect Ratio")
+output_mouth = gr.Textbox(label="Mouth Aspect Ratio")
 
 
 iface = gr.Interface(

model/data/data_download_instructions.md

@@ -0,0 +1,5 @@
+1. Run the following command.
+
+```bash
+python process_data.py
+```