thinh111 commited on
Commit
147ad2f
β€’
1 Parent(s): b1e1753

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -1,4 +1,6 @@
1
- device = "cpu"
 
 
2
  import subprocess
3
  import sys
4
  import gradio as gr
 
1
+ import os
2
+ os.system('export CUDA_VISIBLE_DEVICES=""')
3
+
4
  import subprocess
5
  import sys
6
  import gradio as gr