adding cuda support, first test commit

nemaquant.py

@@ -6,6 +6,7 @@ import numpy as np
 import pandas as pd 
 import cv2
 import os
+from torch import cuda
 from pathlib import Path
 from ultralytics import YOLO
 from glob import glob
@@ -142,6 +143,11 @@ def main():
     if args.key:
         key = parse_key_file(str(args.keypath))
     model = YOLO(args.weights)
+    if cuda.is_available():
+        device = 'cuda'
+    else:
+        device = 'cpu'
+    model.to(device)
     # create a couple empty lists for holding results, easier than adding to empty Pandas DF
     tmp_well = []
     tmp_numeggs = []