home commited on
Commit
10fadcd
·
1 Parent(s): 54cd1ba

Fix ModuleNotFoundError by adding defake to sys.path

Browse files
Files changed (1) hide show
  1. app_infer.py +10 -0
app_infer.py CHANGED
@@ -1,3 +1,13 @@
 
 
 
 
 
 
 
 
 
 
1
  from PIL import Image
2
  import tempfile
3
  from defake.test_api import load_models, predict_image
 
1
+ import sys
2
+ import os
3
+
4
+ # ================= 关键修改开始 =================
5
+ # 把 defake 目录加入到系统路径,解决 ModuleNotFoundError
6
+ current_dir = os.path.dirname(os.path.abspath(__file__))
7
+ defake_dir = os.path.join(current_dir, "defake")
8
+ sys.path.append(defake_dir)
9
+ # ================= 关键修改结束 =================
10
+
11
  from PIL import Image
12
  import tempfile
13
  from defake.test_api import load_models, predict_image