Spaces:
Running
on
Zero
Running
on
Zero
elucidator8918
commited on
Commit
•
1dc8e38
1
Parent(s):
1238702
Update app.py
Browse files
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=
|
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"))
|