elucidator8918 commited on
Commit
1dc8e38
1 Parent(s): 1238702

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ def draw_progress_bar(frame, progress):
43
  cv2.rectangle(frame, (0, 0), (bar_width, bar_height), (0, 255, 0), -1)
44
  cv2.rectangle(frame, (0, 0), (frame_width, bar_height), (255, 255, 255), 1)
45
 
46
- @spaces.GPU(duration=60)
47
  def predict_drowsiness(video_path):
48
  import tensorflow as tf
49
  print(tf.config.list_physical_devices("GPU"))
 
43
  cv2.rectangle(frame, (0, 0), (bar_width, bar_height), (0, 255, 0), -1)
44
  cv2.rectangle(frame, (0, 0), (frame_width, bar_height), (255, 255, 255), 1)
45
 
46
+ @spaces.GPU(duration=80)
47
  def predict_drowsiness(video_path):
48
  import tensorflow as tf
49
  print(tf.config.list_physical_devices("GPU"))