kishan-1721 commited on
Commit
85e9d26
1 Parent(s): 5e7cf99

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = "0", save = True, conf = 0.7)
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')