diff --git a/app/controller/AlgorithmController.py b/app/controller/AlgorithmController.py
index cc79d94..5cb386c 100644
--- a/app/controller/AlgorithmController.py
+++ b/app/controller/AlgorithmController.py
@@ -21,7 +21,7 @@ from app.utils.websocket_tool import manager
 import sys
 from pathlib import Path
 
-# from pynvml import *
+from pynvml import *
 # FILE = Path(__file__).resolve()
 # ROOT = FILE.parents[0]  # YOLOv5 root directory
 # if str(ROOT) not in sys.path: