atalaydenknalbant commited on
Commit
d7b7455
·
verified ·
1 Parent(s): 69c4230

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -17,7 +17,7 @@ def get_model_filenames(repo_id, file_extension = ".pt"):
17
  model_filenames = [file for file in files if file.endswith(file_extension)]
18
  return model_filenames
19
 
20
- repo_id = "atalaydenknalbant/asl-models"
21
  model_filenames = get_model_filenames(repo_id)
22
  print("Model filenames:", model_filenames)
23
 
@@ -36,7 +36,7 @@ def yolo_inference(image, model_id, conf_threshold, iou_threshold, max_detection
36
  model = YOLOv10(model_path)
37
  else:
38
  model = YOLO(model_path)
39
- results = model(source=image, imgsz=416, iou=iou_threshold, conf=conf_threshold, verbose=False, max_det=max_detection)[0]
40
  detections = sv.Detections.from_ultralytics(results)
41
 
42
  labels = [
 
17
  model_filenames = [file for file in files if file.endswith(file_extension)]
18
  return model_filenames
19
 
20
+ repo_id = "atalaydenknalbant/asl-yolo-models"
21
  model_filenames = get_model_filenames(repo_id)
22
  print("Model filenames:", model_filenames)
23
 
 
36
  model = YOLOv10(model_path)
37
  else:
38
  model = YOLO(model_path)
39
+ results = model(source=image, imgsz=640, iou=iou_threshold, conf=conf_threshold, verbose=False, max_det=max_detection)[0]
40
  detections = sv.Detections.from_ultralytics(results)
41
 
42
  labels = [