Spaces:

elucidator8918
/

Drowsiness

Sleeping

elucidator8918 commited on Sep 25, 2024

Commit

2d8937f

verified ·

1 Parent(s): 4aac759

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ from tensorflow.keras.models import load_model
 import gradio as gr
 import tempfile
 import os
 # Load your pre-trained model
 model = load_model('cnn_lstm1.h5')

 import gradio as gr
 import tempfile
 import os
+os.environ['CUDA_VISIBLE_DEVICES'] = "0"
 # Load your pre-trained model
 model = load_model('cnn_lstm1.h5')