Spaces:
Sleeping
Sleeping
Nikhitha2310
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -18,9 +18,9 @@ st.write(
|
|
18 |
|
19 |
status_placeholder = st.empty()
|
20 |
if radio_button1 == "model_train_17":
|
21 |
-
model = YOLO(
|
22 |
elif radio_button1 == "model_train_15":
|
23 |
-
model = YOLO(
|
24 |
|
25 |
|
26 |
|
|
|
18 |
|
19 |
status_placeholder = st.empty()
|
20 |
if radio_button1 == "model_train_17":
|
21 |
+
model = YOLO('train_17_best.pt')
|
22 |
elif radio_button1 == "model_train_15":
|
23 |
+
model = YOLO('train_15_best.pt')
|
24 |
|
25 |
|
26 |
|