Spaces:

kishan-1721
/

object-detection

Sleeping

kishan-1721 commited on Jul 22, 2023

Commit

79ded9c

•

1 Parent(s): dfd026d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,5 +11,5 @@ save_image()
 if st.button("Detect Object"):
     model.predict(source = 'processed_image.jpg', save = True, conf = 0.7)
-    st.image('runs/detect/predict/processed_image.jpg')
     shutil.rmtree('runs')

 if st.button("Detect Object"):
     model.predict(source = 'processed_image.jpg', save = True, conf = 0.7)
+    st.image('runs/detect/predict/processed_image.jpg', use_column_width = 'always')
     shutil.rmtree('runs')