Use setattr magic to fix NeuralNet pickle error elegantly

app.py

@@ -3,6 +3,11 @@ from PIL import Image
 import os
 import sys
 
+current_dir = os.path.dirname(os.path.abspath(__file__))
+defake_dir = os.path.join(current_dir, "defake")
+if defake_dir not in sys.path:
+    sys.path.insert(0, defake_dir)
+
 import defake.test_api
 # 把 defake.test_api 里的 NeuralNet 类，“挂载”到当前的主模块 (__main__) 上
 setattr(sys.modules['__main__'], 'NeuralNet', defake.test_api.NeuralNet)