Update app.py

app.py

@@ -4,10 +4,6 @@ import numpy as np
 from tensorflow.keras.models import load_model
 import gradio as gr
 import tempfile
-import os
-os.environ['CUDA_VISIBLE_DEVICES'] = "0"
-# Load your pre-trained model
-model = load_model('cnn.keras')
 
 # Function to preprocess each frame
 def preprocess_frame(frame):
@@ -39,6 +35,7 @@ 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)