Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,7 @@ global repo_id
|
|
| 9 |
|
| 10 |
repo_id = "atalaydenknalbant/asl-yolo-models"
|
| 11 |
|
| 12 |
-
|
| 13 |
model_filenames = [
|
| 14 |
"yolov10s.pt",
|
| 15 |
"yolov10x.pt",
|
|
@@ -29,7 +29,7 @@ category_dict = {0: 'A', 1: 'B', 2: 'C', 3: 'D', 4: 'E', 5: 'F', 6: 'G', 7: 'H',
|
|
| 29 |
9: 'J', 10: 'K', 11: 'L', 12: 'M', 13: 'N', 14: 'O', 15: 'P', 16: 'Q',
|
| 30 |
17: 'R', 18: 'S', 19: 'T', 20: 'U', 21: 'V', 22: 'W', 23: 'X', 24: 'Y', 25: 'Z'}
|
| 31 |
|
| 32 |
-
@spaces.GPU
|
| 33 |
def yolo_inference(image, model_id, conf_threshold, iou_threshold, max_detection):
|
| 34 |
# Download models
|
| 35 |
model_path = download_models(repo_id, model_id)
|
|
|
|
| 9 |
|
| 10 |
repo_id = "atalaydenknalbant/asl-yolo-models"
|
| 11 |
|
| 12 |
+
|
| 13 |
model_filenames = [
|
| 14 |
"yolov10s.pt",
|
| 15 |
"yolov10x.pt",
|
|
|
|
| 29 |
9: 'J', 10: 'K', 11: 'L', 12: 'M', 13: 'N', 14: 'O', 15: 'P', 16: 'Q',
|
| 30 |
17: 'R', 18: 'S', 19: 'T', 20: 'U', 21: 'V', 22: 'W', 23: 'X', 24: 'Y', 25: 'Z'}
|
| 31 |
|
| 32 |
+
@spaces.GPU(duration=200)
|
| 33 |
def yolo_inference(image, model_id, conf_threshold, iou_threshold, max_detection):
|
| 34 |
# Download models
|
| 35 |
model_path = download_models(repo_id, model_id)
|