Update app.py

app.py

@@ -7,16 +7,6 @@ import time
 import streamlit as st
 import pygame
 
-# Setting up config for GPU usage
-physical_devices = tf.config.list_physical_devices("GPU")
-tf.config.experimental.set_memory_growth(physical_devices[0], True)
-
-# Using Har-cascade classifier from OpenCV
-face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + 'haarcascade_frontalface_default.xml')
-
-# Loading the trained model for prediction purpose
-model = keras.models.load_model('my_model (1).h5')
-
 # Initialize Pygame mixer
 pygame.mixer.init()
 
@@ -25,6 +15,12 @@ def play_alert_sound():
     pygame.mixer.music.load(pygame.mixer.Sound(pygame.mixer.get_init()))  # Load a simple beep sound
     pygame.mixer.music.play()
 
+# Using Har-cascade classifier from OpenCV
+face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + 'haarcascade_frontalface_default.xml')
+
+# Loading the trained model for prediction purpose
+model = keras.models.load_model('my_model (1).h5')
+
 # Title for GUI
 st.title('Drowsiness Detection')
 
@@ -54,10 +50,11 @@ if nav_choice == 'Home':
                 'dataset used is available <a href="https://www.kaggle.com/kutaykutlu/drowsiness-detection", '
                 'target="_blank">here</a></font></b>'
                 , unsafe_allow_html=True)
-
+    
 # Sleep Detection page
 elif nav_choice == 'Sleep Detection':
     st.header('Image Prediction')
+    cap = 0
     st.success('Please look at your web-cam, while following all the instructions given on the Home page.')
     st.warning(
         'Keeping the eyes in the same state is important but you can obviously blink your eyes, if they are open!!!')
@@ -106,6 +103,7 @@ elif nav_choice == 'Sleep Detection':
                 img_container[0].image(frame1, width=250)
                 img_container[2].image(cv2.cvtColor(eye_1, cv2.COLOR_BGR2RGB), width=150)
                 img_container[3].image(cv2.cvtColor(eye_2, cv2.COLOR_BGR2RGB), width=150)
+                print(e1, e2)
                 
                 # Decision variable for prediction
                 if e1 == 1 or e2 == 1:
@@ -126,10 +124,11 @@ elif nav_choice == 'Sleep Detection':
             finally:
                 cap.release()
 
-        # If found drowsy, play the alert sound to alert the driver
+        # If found drowsy, then make a beep sound to alert the driver
         if decision == 0:
             st.error('Eye(s) are closed')
             play_alert_sound()
+
         else:
             st.success('Eyes are Opened')
         st.warning('Please select "Stop" and then "Start" to try again')