mariamyasir commited on
Commit
40e23b3
·
verified ·
1 Parent(s): d541609

Update streamlit_app.py

Browse files
Files changed (1) hide show
  1. streamlit_app.py +1 -1
streamlit_app.py CHANGED
@@ -62,7 +62,7 @@ if image_data:
62
  results = yolo_model(frame, conf=conf_threshold, iou=iou_threshold, imgsz=640)
63
  res = results[0]
64
  annotated_frame = res.plot()
65
- st.image(annotated_frame, caption="Detected Banknotes", use_container_width=True)
66
  try:
67
  xyxy = res.boxes.xyxy.cpu().numpy()
68
  for i, box in enumerate(xyxy):
 
62
  results = yolo_model(frame, conf=conf_threshold, iou=iou_threshold, imgsz=640)
63
  res = results[0]
64
  annotated_frame = res.plot()
65
+ st.image(annotated_frame, caption="Detected Banknotes", width='stretch')
66
  try:
67
  xyxy = res.boxes.xyxy.cpu().numpy()
68
  for i, box in enumerate(xyxy):