diff --git a/core/predict.py b/core/predict.py
index 59edb44..8dde125 100644
--- a/core/predict.py
+++ b/core/predict.py
@@ -4,6 +4,7 @@
 from pathlib import Path
 
 import numpy as np
+import numpy.typing
 import torch
 import logging