dgarrett-synaptics commited on
Commit
59d6b9b
·
1 Parent(s): 1adf951

Adding the correct installation of SyNET

Browse files
Files changed (2) hide show
  1. app.py +4 -28
  2. requirements.txt +1 -9
app.py CHANGED
@@ -1,40 +1,16 @@
1
  import gradio as gr
2
- import os
3
  import spaces
4
  import sys
5
- import subprocess
6
-
7
- for path in sys.path:
8
- print(f'path = {path}')
9
-
10
- # Check isntalled packages
11
- import pkg_resources
12
- installed_packages = pkg_resources.working_set
13
- installed_packages_list = sorted(["%s==%s" % (i.key, i.version)
14
- for i in installed_packages])
15
- print(installed_packages_list)
16
-
17
 
18
  try:
19
  from synet.backends import get_backend
20
  get_backend('ultralytics').patch()
21
  except ImportError:
22
- print('Installed synet_package')
23
- #
24
- #os.system("pip install -e ./synet_package[ultra]")
25
  subprocess.check_call([sys.executable, "-m", "pip", "install", "./synet_package[ultra]"])
26
-
27
-
28
-
29
- installed_packages = pkg_resources.working_set
30
- installed_packages_list = sorted(["%s==%s" % (i.key, i.version)
31
- for i in installed_packages])
32
- print(installed_packages_list)
33
-
34
- files = os.listdir('/usr/local/lib/python3.10/site-packages')
35
- for file in files:
36
- print(f'file = {file}')
37
-
38
  from synet.backends import get_backend
39
  get_backend('ultralytics').patch()
40
 
 
1
  import gradio as gr
 
2
  import spaces
3
  import sys
4
+ import os
 
 
 
 
 
 
 
 
 
 
 
5
 
6
  try:
7
  from synet.backends import get_backend
8
  get_backend('ultralytics').patch()
9
  except ImportError:
10
+ import subprocess
11
+ print('Installing synet package')
 
12
  subprocess.check_call([sys.executable, "-m", "pip", "install", "./synet_package[ultra]"])
13
+ # Import the file
 
 
 
 
 
 
 
 
 
 
 
14
  from synet.backends import get_backend
15
  get_backend('ultralytics').patch()
16
 
requirements.txt CHANGED
@@ -1,12 +1,4 @@
1
  gradio
2
  torch
3
  tensorflow
4
- ultralytics
5
- #transformers
6
- #datasets
7
- #trl
8
- #peft
9
- #tf-keras
10
- #bitsandbytes
11
- #pytest
12
- #/tmp/synet_package[ultra]
 
1
  gradio
2
  torch
3
  tensorflow
4
+ ultralytics