Merge pull request #150 from JLWLL/master

update cartoon-sd and give aihub server debug choose

aihub/deep-learning/cartoon-sd/app.py

@@ -3,6 +3,11 @@ import io, sys, os
 __dir__ = os.path.dirname(os.path.abspath(__file__))
 sys.path.append(os.path.abspath(os.path.join(__dir__, '/naifu')))
 
+try:
+    os.system("ln -f -s /naifu/* /app/")
+except:
+    print("软连接出错，请重试！")
+
 import random
 from datetime import datetime
 from cubestudio.aihub.model import Model
@@ -141,6 +146,7 @@ class Cartoon_SD_Model(Model):
         try:
             if optim.seed is None:
                 optim.seed = random.randint(0, 100000000)
+            optim.mitigate = False
             images = self.model.sample(optim)
             ndarray_convert_img_list = []
             for image in images:
@@ -192,4 +198,4 @@ model = Cartoon_SD_Model()
 
 # 启动服务
 server = Server(model=model)
-server.server(port=8080)
+server.server(port=8080,debug=False)

aihub/src/cubestudio/aihub/web/server.py

@@ -67,7 +67,7 @@ class Server():
 
     # 启动服务
     # @pysnooper.snoop()
-    def server(self,port=8080):
+    def server(self, port=8080, debug=True):
 
         app = Flask(__name__,
                     static_url_path=f'/{self.pre_url}/static/',
@@ -500,7 +500,7 @@ class Server():
                 print(user_history)
             return response
 
-        app.run(host='0.0.0.0', debug=True, port=port)
+        app.run(host='0.0.0.0', debug=debug, port=port)