diff --git a/serving/sf2f/inference.py b/serving/sf2f/inference.py
index 5522c1a..7452a67 100644
--- a/serving/sf2f/inference.py
+++ b/serving/sf2f/inference.py
@@ -1,3 +1,4 @@
+import .models
 import torch
 import os, glob
 import mlflow