lawrence722 commited on
Commit
3a5d909
1 Parent(s): 76c799d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = "./models-20240613T085418Z-001"
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("models-20240613T085418Z-001", ["Upload Image", "Image URL"])
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.")