update
Browse files
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 |
-
|
34 |
-
|
35 |
-
|
36 |
-
|
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 |
|