ch-tseng commited on
Commit
4659419
·
1 Parent(s): 4f70fa6
Files changed (1) hide show
  1. app.py +14 -14
app.py CHANGED
@@ -18,6 +18,15 @@ import config
18
  from utils import load_model, infer_uploaded_image, infer_uploaded_video, infer_uploaded_webcam
19
  import os
20
 
 
 
 
 
 
 
 
 
 
21
  base_download_path = "downloaded"
22
  #------------------------
23
  if not os.path.exists(base_download_path):
@@ -29,12 +38,11 @@ model_count = int(st.secrets["model_count"])
29
  model_info = {}
30
  models_list = []
31
  for i in range(0, model_count):
32
- '''
33
- model_name = os.getenv("m{}_name".format(i))
34
- gdrive_id = os.getenv("m{}_griv".format(i))
35
- model_extname = os.getenv("m{}_type".format(i))
36
- model_desc = os.getenv("m{}_desc".format(i))
37
- '''
38
  model_name = st.secrets["m{}_name".format(i)]
39
  gdrive_id = st.secrets["m{}_griv".format(i)]
40
  model_extname = st.secrets["m{}_type".format(i)]
@@ -57,14 +65,6 @@ for i in range(0, model_count):
57
 
58
  print('models_list', models_list)
59
 
60
- # setting page layout
61
- st.set_page_config(
62
- page_title="YOLO.dog",
63
- page_icon="🤖",
64
- layout="wide",
65
- initial_sidebar_state="expanded"
66
- )
67
-
68
  # main page heading
69
  st.title("Models Demo")
70
 
 
18
  from utils import load_model, infer_uploaded_image, infer_uploaded_video, infer_uploaded_webcam
19
  import os
20
 
21
+ # setting page layout
22
+ st.set_page_config(
23
+ page_title="YOLO.dog",
24
+ page_icon="🤖",
25
+ layout="wide",
26
+ initial_sidebar_state="expanded"
27
+ )
28
+
29
+
30
  base_download_path = "downloaded"
31
  #------------------------
32
  if not os.path.exists(base_download_path):
 
38
  model_info = {}
39
  models_list = []
40
  for i in range(0, model_count):
41
+ #model_name = os.getenv("m{}_name".format(i))
42
+ #gdrive_id = os.getenv("m{}_griv".format(i))
43
+ #model_extname = os.getenv("m{}_type".format(i))
44
+ #model_desc = os.getenv("m{}_desc".format(i))
45
+
 
46
  model_name = st.secrets["m{}_name".format(i)]
47
  gdrive_id = st.secrets["m{}_griv".format(i)]
48
  model_extname = st.secrets["m{}_type".format(i)]
 
65
 
66
  print('models_list', models_list)
67
 
 
 
 
 
 
 
 
 
68
  # main page heading
69
  st.title("Models Demo")
70