diff --git a/aihub/deep-learning/cartoon-sd/app.py b/aihub/deep-learning/cartoon-sd/app.py index 5fec5455..09509e5e 100644 --- a/aihub/deep-learning/cartoon-sd/app.py +++ b/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) diff --git a/aihub/src/cubestudio/aihub/web/server.py b/aihub/src/cubestudio/aihub/web/server.py index 9a855053..76b30047 100644 --- a/aihub/src/cubestudio/aihub/web/server.py +++ b/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)