mohamedrekik commited on
Commit
18d4e5f
·
1 Parent(s): f03af75

cast model_ind type

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -108,7 +108,7 @@ model_names = ["yolov8n 640 mask_300000_3000_0.3_0.7.onnx",
108
  input_size = 640
109
 
110
 
111
- def run(img_path, model_ind):
112
  image_data, img_width, img_height, original_image = image_preprocess(img_path)
113
  model_name = model_names[model_ind]
114
  detections = inference(model_name, image_data)
 
108
  input_size = 640
109
 
110
 
111
+ def run(img_path, model_ind:int):
112
  image_data, img_width, img_height, original_image = image_preprocess(img_path)
113
  model_name = model_names[model_ind]
114
  detections = inference(model_name, image_data)