Yuxiang Wang commited on
Commit
e01f7c2
·
1 Parent(s): 674635a

bypass disk memory limit for pkg installation

Browse files
Files changed (1) hide show
  1. app.py +6 -1
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
- subprocess.call(f"pip install {package_spec}".split())
 
 
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: