andy-wyx commited on
Commit
f1c6a3d
·
1 Parent(s): 1d7c63d

chore:create new branch,run local pkg installation

Browse files
Files changed (2) hide show
  1. .gitignore +2 -0
  2. app.py +45 -1
.gitignore CHANGED
@@ -1,3 +1,5 @@
 
 
1
  .env
2
  venv/
3
  images/
 
1
+ node_modules/
2
+ *.json
3
  .env
4
  venv/
5
  images/
app.py CHANGED
@@ -1,6 +1,49 @@
1
 
2
  import subprocess
3
- import os
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  if os.getenv('SYSTEM') == 'spaces':
5
  subprocess.call('pip install tensorflow==2.9'.split())
6
  subprocess.call('pip install keras==2.9'.split())
@@ -10,6 +53,7 @@ if os.getenv('SYSTEM') == 'spaces':
10
  subprocess.call('pip install python-dotenv'.split())
11
  subprocess.call('pip install torch torchvision '.split())
12
  subprocess.call('pip install xplique'.split())
 
13
 
14
  import gradio as gr
15
  from huggingface_hub import snapshot_download
 
1
 
2
  import subprocess
3
+ import os
4
+ import sys
5
+ import pkg_resources
6
+ '''
7
+ env_name = os.path.basename(sys.prefix)
8
+ if env_name == 'fossil':
9
+ subprocess.call('pip install tensorflow==2.9'.split())
10
+ subprocess.call('pip install keras==2.9'.split())
11
+ subprocess.call('pip install git+https://github.com/facebookresearch/segment-anything.git'.split())
12
+ subprocess.call('pip install opencv-python-headless==4.5.5.64'.split())
13
+ subprocess.call('pip install git+https://github.com/cocodataset/panopticapi.git'.split())
14
+ subprocess.call('pip install python-dotenv'.split())
15
+ subprocess.call('pip install torch torchvision '.split())
16
+ subprocess.call('pip install xplique'.split())
17
+ subprocess.call('pip install gradio'.split())
18
+ else:
19
+ print("Please use venv named 'fossil' in case pkgs installed to other unexpected envs or locations.")
20
+
21
+ def is_pkg_installed(pkg_name):
22
+ installed_pkgs = {pkg.key for pkg in pkg_resources.working_set}
23
+ return pkg_name.lower() in installed_pkgs
24
+
25
+ packages_to_install = [
26
+ ('tensorflow', '2.9'),
27
+ ('keras', '2.9'),
28
+ ('opencv-python-headless', '4.5.5.64'),
29
+ ('python-dotenv', None),
30
+ ('torch', None),
31
+ ('torchvision', None),
32
+ ('xplique', None),
33
+ ]
34
+
35
+ for package, version in packages_to_install:
36
+ package_spec = f"{package}=={version}" if version else package
37
+ if not is_pkg_installed(package):
38
+ print(f"Installing {package_spec}...")
39
+ subprocess.call(f"pip install {package_spec}".split())
40
+ else:
41
+ print(f"{package_spec} is already installed.")
42
+
43
+ #system_setting = os.getenv('SYSTEM', 'spaces')
44
+ #print(system_setting)
45
+ '''
46
+ '''
47
  if os.getenv('SYSTEM') == 'spaces':
48
  subprocess.call('pip install tensorflow==2.9'.split())
49
  subprocess.call('pip install keras==2.9'.split())
 
53
  subprocess.call('pip install python-dotenv'.split())
54
  subprocess.call('pip install torch torchvision '.split())
55
  subprocess.call('pip install xplique'.split())
56
+ '''
57
 
58
  import gradio as gr
59
  from huggingface_hub import snapshot_download