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')
|