mart9992
/

theimageconvert2

Model card Files Files and versions

mart9992 commited on Jan 6, 2024

Commit

389b1cb

·

1 Parent(s): 4c65bff

m

Files changed (1) hide show

handler.py +2 -2

handler.py CHANGED Viewed

@@ -6,8 +6,8 @@ from scipy.ndimage import binary_dilation
 import subprocess, io, os, sys, time
-is_production = False
-install_stuff = False
 os.environ['CUDA_HOME'] = '/usr/local/cuda-11.7/' if is_production else '/usr/local/cuda-12.1/'
 run_gradio = False

 import subprocess, io, os, sys, time
+is_production = True
+install_stuff = True
 os.environ['CUDA_HOME'] = '/usr/local/cuda-11.7/' if is_production else '/usr/local/cuda-12.1/'
 run_gradio = False