Spaces:
Sleeping
Sleeping
update pre-installed dependencies
Browse files- env.py +1 -9
- requirements.txt +1 -0
env.py
CHANGED
@@ -22,9 +22,6 @@ def config_env():
|
|
22 |
|
23 |
name_to_command = {'segment_anything':'git+https://github.com/facebookresearch/segment-anything.git',
|
24 |
'panopticapi':'git+https://github.com/cocodataset/panopticapi.git',
|
25 |
-
#TODO
|
26 |
-
'torch':'torch --index-url https://download.pytorch.org/whl/cu118',
|
27 |
-
'torchvision':'torchvision --index-url https://download.pytorch.org/whl/cu118',
|
28 |
}
|
29 |
|
30 |
env_name = os.path.basename(sys.prefix)
|
@@ -33,12 +30,7 @@ def config_env():
|
|
33 |
package_spec = f"{package}=={version}" if version else package
|
34 |
package_spec = name_to_command[package_spec] if package_spec in name_to_command else package_spec
|
35 |
if not is_pkg_installed(package):
|
36 |
-
|
37 |
-
if package=='torch' or 'torchvision' or 'tensorflow':
|
38 |
-
# TMPDIR = 'home/username/tmp' for temporary compilation to prevent disk partition storage limit on zmachine
|
39 |
-
subprocess.call(f"TMPDIR='../../../tmp/' pip install {package_spec}",shell=True)
|
40 |
-
else:
|
41 |
-
subprocess.call(f"pip install {package_spec}".split())
|
42 |
else:
|
43 |
print(f"{package_spec} is already installed.")
|
44 |
else:
|
|
|
22 |
|
23 |
name_to_command = {'segment_anything':'git+https://github.com/facebookresearch/segment-anything.git',
|
24 |
'panopticapi':'git+https://github.com/cocodataset/panopticapi.git',
|
|
|
|
|
|
|
25 |
}
|
26 |
|
27 |
env_name = os.path.basename(sys.prefix)
|
|
|
30 |
package_spec = f"{package}=={version}" if version else package
|
31 |
package_spec = name_to_command[package_spec] if package_spec in name_to_command else package_spec
|
32 |
if not is_pkg_installed(package):
|
33 |
+
subprocess.call(f"pip install {package_spec}".split())
|
|
|
|
|
|
|
|
|
|
|
34 |
else:
|
35 |
print(f"{package_spec} is already installed.")
|
36 |
else:
|
requirements.txt
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
subprocess
|
2 |
importlib.metadata
|
3 |
gradio
|
|
|
1 |
+
python==3.9.12
|
2 |
subprocess
|
3 |
importlib.metadata
|
4 |
gradio
|