truthdotphd commited on
Commit
c0035e7
·
verified ·
1 Parent(s): 36d88d2

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +26 -0
model.py CHANGED
@@ -1,3 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import numpy as np
2
  import triton_python_backend_utils as pb_utils
3
  from omnicloudmask import predict_from_array
 
1
+ import os
2
+ import sys
3
+ import subprocess
4
+
5
+ # Get script's absolute directory path
6
+ script_dir = os.path.dirname(os.path.abspath(__file__))
7
+ requirements_path = os.path.join(script_dir, "requirements.txt")
8
+
9
+ # Install with path validation
10
+ try:
11
+ subprocess.check_call([
12
+ sys.executable,
13
+ "-m",
14
+ "pip",
15
+ "install",
16
+ "-r",
17
+ requirements_path
18
+ ])
19
+ except subprocess.CalledProcessError as e:
20
+ print(f"Installation failed with error code {e.returncode}")
21
+ sys.exit(1)
22
+ except FileNotFoundError:
23
+ print(f"requirements.txt not found at: {requirements_path}")
24
+ sys.exit(1)
25
+
26
+
27
  import numpy as np
28
  import triton_python_backend_utils as pb_utils
29
  from omnicloudmask import predict_from_array