Update task_service.py

This commit is contained in:
Edwin Jose 2025-09-08 13:17:24 -04:00
parent 975a4c214a
commit 87e31d98a4

View file

@ -1,6 +1,7 @@
import asyncio import asyncio
import random import random
from typing import Dict, Optional from typing import Dict, Optional
import uuid
from models.tasks import TaskStatus, UploadTask, FileTask from models.tasks import TaskStatus, UploadTask, FileTask
from utils.gpu_detection import get_worker_count from utils.gpu_detection import get_worker_count