diff --git a/ieemoo-ai-searchv2.py b/ieemoo-ai-searchv2.py index da1b4a2..5ce64ee 100755 --- a/ieemoo-ai-searchv2.py +++ b/ieemoo-ai-searchv2.py @@ -25,7 +25,7 @@ if SW_SERVER and SW_SERVICE_NAME: config.init() #采集服务的地址,给自己的服务起个名称 #config.init(collector="123.60.56.51:11800", service='ieemoo-ai-search') #采集服务的地址,给自己的服务起个名称 agent.start() -os.environ["CUDA_VISIBLE_DEVICES"] = '0' +os.environ["CUDA_VISIBLE_DEVICES"] = '1' app = Flask(__name__) @@ -58,7 +58,7 @@ def setup_logging(path): return logger logger = setup_logging('utils/logging.json') -@app.route('/searchv2', methods=['POST']) +@app.route('/searchv2', methods=['POST', 'GET']) def search(): pre_status = False try: