fix: pass extra_search path (#687)

jina-ai · Apr 21, 2022 · 2558d73 · 2558d73
1 parent e3d4e91
commit 2558d73
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/server/clip_server/__main__.py b/server/clip_server/__main__.py
@@ -1,8 +1,12 @@
 import sys
-
+import os
+import inspect
 from jina import Flow
 
 if __name__ == '__main__':
-    f = Flow.load_config('torch-flow.yml' if len(sys.argv) == 1 else sys.argv[1])
+    f = Flow.load_config(
+        'torch-flow.yml' if len(sys.argv) == 1 else sys.argv[1],
+        extra_search_paths=[os.path.dirname(inspect.getfile(inspect.currentframe()))],
+    )
     with f:
         f.block()