IliaLarchenko
commited on
Commit
•
5aca4ad
1
Parent(s):
93c9480
refactoring
Browse files- src/app.py +3 -3
src/app.py
CHANGED
@@ -2,8 +2,6 @@ import os
|
|
2 |
import streamlit as st
|
3 |
import albumentations as A
|
4 |
|
5 |
-
# TODO: refactor all the new code of professional mode
|
6 |
-
|
7 |
from utils import load_augmentations_config, get_arguments
|
8 |
from visuals import (
|
9 |
show_transform_control,
|
@@ -12,6 +10,8 @@ from visuals import (
|
|
12 |
show_docstring,
|
13 |
)
|
14 |
|
|
|
|
|
15 |
# get CLI params: the path to images and image width
|
16 |
path_to_images, width_original = get_arguments()
|
17 |
if not os.path.isdir(path_to_images):
|
@@ -19,7 +19,7 @@ if not os.path.isdir(path_to_images):
|
|
19 |
else:
|
20 |
# select interface type
|
21 |
interface_type = st.sidebar.radio(
|
22 |
-
"Select the interface
|
23 |
)
|
24 |
|
25 |
# select image
|
|
|
2 |
import streamlit as st
|
3 |
import albumentations as A
|
4 |
|
|
|
|
|
5 |
from utils import load_augmentations_config, get_arguments
|
6 |
from visuals import (
|
7 |
show_transform_control,
|
|
|
10 |
show_docstring,
|
11 |
)
|
12 |
|
13 |
+
# TODO: refactor all the new code of professional mode
|
14 |
+
|
15 |
# get CLI params: the path to images and image width
|
16 |
path_to_images, width_original = get_arguments()
|
17 |
if not os.path.isdir(path_to_images):
|
|
|
19 |
else:
|
20 |
# select interface type
|
21 |
interface_type = st.sidebar.radio(
|
22 |
+
"Select the interface mode", ["Simple", "Professional"]
|
23 |
)
|
24 |
|
25 |
# select image
|