Spaces:
Sleeping
Sleeping
home
commited on
Commit
·
10fadcd
1
Parent(s):
54cd1ba
Fix ModuleNotFoundError by adding defake to sys.path
Browse files- 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
|