- handler.py +2 -2
handler.py
CHANGED
@@ -6,8 +6,8 @@ from scipy.ndimage import binary_dilation
|
|
6 |
|
7 |
import subprocess, io, os, sys, time
|
8 |
|
9 |
-
is_production =
|
10 |
-
install_stuff =
|
11 |
os.environ['CUDA_HOME'] = '/usr/local/cuda-11.7/' if is_production else '/usr/local/cuda-12.1/'
|
12 |
|
13 |
run_gradio = False
|
|
|
6 |
|
7 |
import subprocess, io, os, sys, time
|
8 |
|
9 |
+
is_production = True
|
10 |
+
install_stuff = True
|
11 |
os.environ['CUDA_HOME'] = '/usr/local/cuda-11.7/' if is_production else '/usr/local/cuda-12.1/'
|
12 |
|
13 |
run_gradio = False
|