From 3e95a9091977a4a1c69e4dcadb99c4ff9e228c77 Mon Sep 17 00:00:00 2001
From: soyoonjeong <76814748+soyoonjeong@users.noreply.github.com>
Date: Thu, 14 Mar 2024 02:35:53 +0900
Subject: [PATCH] Fix : models import error2

---
 serving/sf2f/inference.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/serving/sf2f/inference.py b/serving/sf2f/inference.py
index 7452a67..cca4417 100644
--- a/serving/sf2f/inference.py
+++ b/serving/sf2f/inference.py
@@ -1,4 +1,7 @@
-import .models
+try:
+    from . import models
+except:
+    import .models as models
 import torch
 import os, glob
 import mlflow