Update src/streamlit_app.py

src/streamlit_app.py

@@ -6,6 +6,7 @@ import streamlit as st
 import numpy as np
 import torch
 import time
+import cv2
 from numpy import random
 import os
 import sys
@@ -156,4 +157,4 @@ else:
             pil_img = PILImage.create(BytesIO(response.content))
             detect_modify(pil_img, model, conf=conf, imgsz=imgsz, conf_thres=conf_thres, iou_thres=iou_thres)
         except:
-            st.text("Problem reading image from", url)
+            st.text("Problem reading image from", url)