Update app.py

app.py

@@ -4,6 +4,8 @@ import numpy as np
 from tensorflow.keras.models import load_model
 import gradio as gr
 import tempfile
+print(tf.config.list_physical_devices("GPU"))
+model = load_model('cnn.keras')
 
 # Function to preprocess each frame
 def preprocess_frame(frame):
@@ -35,7 +37,6 @@ def predict_drowsiness(video_path):
         preprocessed_frame = preprocess_frame(frame)
         
         # Use the model to predict drowsiness
-        model = load_model('cnn.keras')
         prediction = model.predict(preprocessed_frame)
         drowsiness = np.argmax(prediction)