diff --git a/core/globals.py b/core/globals.py
index 3525c9d..65d3283 100644
--- a/core/globals.py
+++ b/core/globals.py
@@ -1,3 +1,4 @@
+import torch
 import onnxruntime
 
 use_gpu = False