Update app.py

app.py

@@ -1,4 +1,5 @@
 import cv2
+import spaces
 import numpy as np
 from tensorflow.keras.models import load_model
 import gradio as gr
@@ -14,7 +15,7 @@ def preprocess_frame(frame):
     normalized_frame = resized_frame / 255.0
     return np.expand_dims(normalized_frame, axis=0)  # Add batch dimension
 
-# Function to process the video and make predictions
+@spaces.GPU(duration=120)
 def predict_drowsiness(video_path):
     # Open the video file
     cap = cv2.VideoCapture(video_path)