Spaces:
Sleeping
Sleeping
Commit
•
3a5d909
1
Parent(s):
76c799d
Update app.py
Browse files
app.py
CHANGED
@@ -118,7 +118,7 @@ conf = 0.4
|
|
118 |
conf_thres = 0.25
|
119 |
iou_thres=0.45
|
120 |
device = torch.device("cpu")
|
121 |
-
path = "./
|
122 |
|
123 |
# Load model
|
124 |
model = attempt_load(weight_path, map_location=torch.device('cpu')) # load FP32 model
|
@@ -127,7 +127,7 @@ model = attempt_load(weight_path, map_location=torch.device('cpu')) # load FP32
|
|
127 |
# YOLOv7
|
128 |
This is a object detection model for [Objects].
|
129 |
"""
|
130 |
-
option = st.radio("
|
131 |
|
132 |
if option == "Upload Image":
|
133 |
uploaded_file = st.file_uploader("Please upload an image.")
|
|
|
118 |
conf_thres = 0.25
|
119 |
iou_thres=0.45
|
120 |
device = torch.device("cpu")
|
121 |
+
path = "./"
|
122 |
|
123 |
# Load model
|
124 |
model = attempt_load(weight_path, map_location=torch.device('cpu')) # load FP32 model
|
|
|
127 |
# YOLOv7
|
128 |
This is a object detection model for [Objects].
|
129 |
"""
|
130 |
+
option = st.radio("", ["Upload Image", "Image URL"])
|
131 |
|
132 |
if option == "Upload Image":
|
133 |
uploaded_file = st.file_uploader("Please upload an image.")
|