h2chen commited on
Commit
5eeffec
·
verified ·
1 Parent(s): 3d3cddb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 = './yolov7.pt'
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 [Objects].
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