linhcuem commited on
Commit
e440d23
1 Parent(s): c0e60fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ def predict(image, threshold=0.25, model_id=None):
44
  def show_pred_vid(
45
  video_path: str = None,
46
  model_path: str = None,
47
- image_size: 640,
48
  conf_threshold: float = 0.25,
49
  iou_threshold: float = 0.45,
50
  ):
 
44
  def show_pred_vid(
45
  video_path: str = None,
46
  model_path: str = None,
47
+ image_size: int = 640,
48
  conf_threshold: float = 0.25,
49
  iou_threshold: float = 0.45,
50
  ):