Update device support

tasks/image.py

@@ -18,7 +18,7 @@ load_dotenv()
 
 router = APIRouter()
 
-DESCRIPTION = "YOLO-EFD"
+DESCRIPTION = "YOLO-EFD CEA-List"
 ROUTE = "/image"
 
 def parse_boxes(annotation_string):
@@ -96,7 +96,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
     test_dataset = train_test["test"]
 
     model = YOLOv10("best.pt")
-    device = "cuda"
+    device = "cuda" if torch.cuda.is_available() else "cpu"
     model = model.to(device)
     
     # Start tracking emissions