ch-tseng commited on
Commit
a3f3e32
·
1 Parent(s): b61deaf
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -20,17 +20,21 @@ import os
20
  if not os.path.exists('./crowded_human_yolov8s.pt'):
21
  download_file_from_google_drive('1qCXBDy3YuxS9bqfRIc--cRLo1L3DAYq2', './crowded_human_yolov8s.pt')
22
 
 
 
 
 
23
 
24
  # setting page layout
25
  st.set_page_config(
26
- page_title="Interactive Interface for YOLOv8",
27
  page_icon="🤖",
28
  layout="wide",
29
  initial_sidebar_state="expanded"
30
  )
31
 
32
  # main page heading
33
- st.title("Interactive Interface for YOLOv8")
34
 
35
  # sidebar
36
  st.sidebar.header("DL Model Config")
 
20
  if not os.path.exists('./crowded_human_yolov8s.pt'):
21
  download_file_from_google_drive('1qCXBDy3YuxS9bqfRIc--cRLo1L3DAYq2', './crowded_human_yolov8s.pt')
22
 
23
+ if os.path.exists('./crowded_human_yolov8s.pt'):
24
+ file_exist = "./crowded_human_yolov8s.pt"
25
+ else:
26
+ file_exist = "No file"
27
 
28
  # setting page layout
29
  st.set_page_config(
30
+ page_title="YOLO.dog",
31
  page_icon="🤖",
32
  layout="wide",
33
  initial_sidebar_state="expanded"
34
  )
35
 
36
  # main page heading
37
+ st.title(file_exist)
38
 
39
  # sidebar
40
  st.sidebar.header("DL Model Config")