Spaces:
Runtime error
Runtime error
Commit
·
18d4e5f
1
Parent(s):
f03af75
cast model_ind type
Browse files
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)
|