Spaces:
Sleeping
Sleeping
Yuxiang Wang
commited on
Commit
·
e01f7c2
1
Parent(s):
674635a
bypass disk memory limit for pkg installation
Browse files
app.py
CHANGED
@@ -10,6 +10,7 @@ def is_pkg_installed(pkg_name):
|
|
10 |
return True
|
11 |
except importlib.metadata.PackageNotFoundError:
|
12 |
return False
|
|
|
13 |
packages_to_install = [
|
14 |
('gradio',None),
|
15 |
('numpy',None),
|
@@ -20,6 +21,8 @@ packages_to_install = [
|
|
20 |
('torch', None),
|
21 |
('torchvision', None),
|
22 |
('xplique', None),
|
|
|
|
|
23 |
]
|
24 |
|
25 |
env_name = os.path.basename(sys.prefix)
|
@@ -27,7 +30,9 @@ if env_name == 'fossil': # in case pkgs installed to unexpected env during local
|
|
27 |
for package, version in packages_to_install:
|
28 |
package_spec = f"{package}=={version}" if version else package
|
29 |
if not is_pkg_installed(package):
|
30 |
-
|
|
|
|
|
31 |
else:
|
32 |
print(f"{package_spec} is already installed.")
|
33 |
else:
|
|
|
10 |
return True
|
11 |
except importlib.metadata.PackageNotFoundError:
|
12 |
return False
|
13 |
+
|
14 |
packages_to_install = [
|
15 |
('gradio',None),
|
16 |
('numpy',None),
|
|
|
21 |
('torch', None),
|
22 |
('torchvision', None),
|
23 |
('xplique', None),
|
24 |
+
('git+https://github.com/facebookresearch/segment-anything.git', None),
|
25 |
+
('git+https://github.com/cocodataset/panopticapi.git', None),
|
26 |
]
|
27 |
|
28 |
env_name = os.path.basename(sys.prefix)
|
|
|
30 |
for package, version in packages_to_install:
|
31 |
package_spec = f"{package}=={version}" if version else package
|
32 |
if not is_pkg_installed(package):
|
33 |
+
#TODO
|
34 |
+
subprocess.call(f"TMPDIR='../../../tmp/' pip install {package_spec}",shell=True)
|
35 |
+
#subprocess.call(f"pip install {package_spec}".split())
|
36 |
else:
|
37 |
print(f"{package_spec} is already installed.")
|
38 |
else:
|