Spaces:
Sleeping
Sleeping
Commit
•
85e9d26
1
Parent(s):
5e7cf99
Update app.py
Browse files
app.py
CHANGED
@@ -10,6 +10,6 @@ st.title('Object Detection')
|
|
10 |
save_image()
|
11 |
|
12 |
if st.button("Detect Object"):
|
13 |
-
model.predict(source =
|
14 |
st.image('runs/detect/predict/processed_image.jpg', use_column_width = 'always')
|
15 |
shutil.rmtree('runs')
|
|
|
10 |
save_image()
|
11 |
|
12 |
if st.button("Detect Object"):
|
13 |
+
model.predict(source = 'processed_image.jpg', save = True, conf = 0.7)
|
14 |
st.image('runs/detect/predict/processed_image.jpg', use_column_width = 'always')
|
15 |
shutil.rmtree('runs')
|