diff --git a/main_onnx_amd.py b/main_onnx_amd.py
index 4deff1d..2d6c4ff 100644
--- a/main_onnx_amd.py
+++ b/main_onnx_amd.py
@@ -1,6 +1,5 @@
 import onnxruntime as ort
 import numpy as np
-import cupy as cp
 import pyautogui
 import gc
 import numpy as np
@@ -12,6 +11,7 @@ import pandas as pd
 from utils.general import (cv2, non_max_suppression, xyxy2xywh)
 import dxcam
 import torch
+import torch_directml
 
 
 def main():
@@ -109,7 +109,7 @@ def main():
         npImg = npImg.astype(cp.half)
         npImg = cp.moveaxis(npImg, 3, 1)
 
-        outputs = ort_sess.run(None, {'images': cp.asnumpy(npImg)})
+        outputs = ort_sess.run(None, {'images': npImg})
 
         im = torch.from_numpy(outputs[0]).to('cpu')