Update streamlit_app.py
Browse files- 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",
|
| 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):
|