m

handler.py

@@ -7,13 +7,13 @@ print(torch.version.cuda)
 
 class EndpointHandler():
     def __init__(self, path=""):
-        is_production = False
+        is_production = True
         
         os.chdir("/repository" if is_production else "/workspace/eri2")
         
         os.environ['AM_I_DOCKER'] = 'False'
         os.environ['BUILD_WITH_CUDA'] = 'True'
-        os.environ['CUDA_HOME'] = '/usr/local/cuda-11.8/'
+        os.environ['CUDA_HOME'] = '/usr/local/cuda-11.7/'
 
         # Install Segment Anything
         subprocess.run(["python", "-m", "pip", "install", "-e", "segment_anything"])