Update imports for HF inference endpoint

handler.py

@@ -6,10 +6,9 @@ import os
 from io import BytesIO
 import json
 
-from pathlib import Path
-CODE_PATH = Path('code/')
 import sys
-sys.path.append(str(CODE_PATH))
+CODE_PATH = os.path.join(os.path.dirname(os.path.abspath(__file__)), "code")
+sys.path.append(CODE_PATH)
 from clip.model import CLIP
 from clip.clip import _transform, tokenize
 

inference.py

@@ -4,8 +4,11 @@ import base64
 from io import BytesIO
 import json
 import sys
+import os
 
-sys.path.append("code")
+import sys
+CODE_PATH = os.path.join(os.path.dirname(os.path.abspath(__file__)), "code")
+sys.path.append(CODE_PATH)
 from clip.model import CLIP
 from clip.clip import _transform, tokenize