Dricz commited on
Commit
f2e74e9
1 Parent(s): 80fb3d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ video_directory = "/home/user/app/video"
43
 
44
  def response2(image: gr.Image = None,image_size: gr.Slider = 640, conf_threshold: gr.Slider = 0.3, iou_threshold: gr.Slider = 0.6):
45
 
46
- results = model.predict(image, conf=conf_threshold, iou=iou_threshold, imgsz=image_size
47
 
48
  text = ""
49
  name_weap = ""
 
43
 
44
  def response2(image: gr.Image = None,image_size: gr.Slider = 640, conf_threshold: gr.Slider = 0.3, iou_threshold: gr.Slider = 0.6):
45
 
46
+ results = model.predict(image, conf=conf_threshold, iou=iou_threshold, imgsz=image_size)
47
 
48
  text = ""
49
  name_weap = ""