Spaces:

elucidator8918
/

Drowsiness

Sleeping

elucidator8918 commited on Sep 25, 2024

Commit

ea7c227

verified ·

1 Parent(s): 89530ea

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,6 +4,8 @@ import numpy as np
 import tensorflow as tf
 import gradio as gr
 import tempfile
 print(tf.config.list_physical_devices("GPU"))
 model = tf.keras.models.load_model('cnn.keras')

 import tensorflow as tf
 import gradio as gr
 import tempfile
+import os
+os.environ['CUDA_VISIBLE_DEVICES'] = "0"
 print(tf.config.list_physical_devices("GPU"))
 model = tf.keras.models.load_model('cnn.keras')