Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -112,7 +112,7 @@ def detect_modify(img0, model, conf=0.4, imgsz=640, conf_thres = 0.25, iou_thres
|
|
112 |
st.image(img0, caption="Prediction Result", use_column_width=True)
|
113 |
|
114 |
#set paramters
|
115 |
-
weight_path = './
|
116 |
imgsz = 640
|
117 |
conf = 0.4
|
118 |
conf_thres = 0.25
|
@@ -125,7 +125,7 @@ model = attempt_load(weight_path, map_location=torch.device('cpu')) # load FP32
|
|
125 |
|
126 |
"""
|
127 |
# YOLOv7
|
128 |
-
This is a object detection model for [
|
129 |
"""
|
130 |
option = st.radio("", ["Upload Image", "Image URL"])
|
131 |
|
|
|
112 |
st.image(img0, caption="Prediction Result", use_column_width=True)
|
113 |
|
114 |
#set paramters
|
115 |
+
weight_path = './best.pt'
|
116 |
imgsz = 640
|
117 |
conf = 0.4
|
118 |
conf_thres = 0.25
|
|
|
125 |
|
126 |
"""
|
127 |
# YOLOv7
|
128 |
+
This is a object detection model for [Cat].
|
129 |
"""
|
130 |
option = st.radio("", ["Upload Image", "Image URL"])
|
131 |
|