master
/ handler.py

handler.py @e4f03e7 raw · history · blame

from yi import YiProgram

def handle(conf):
    """
    该方法是部署之后,其他人调用你的服务时候的处理方法。
    请按规范填写参数结构,这样我们就能替你自动生成配置文件,方便其他人的调用。
    范例:
    params['key'] = value # value_type: str # description: some description
    value_type 可以选择:img, video, audio, str, int, float, [int], [str], [float]
    参数请放到params字典中,我们会自动解析该变量。
    """

    wish = conf['wish']  # value_type: str # description: some description
    pray = conf['pray']

    obj = YiProgram()
    turn_symbol_desc, main_indicate, support_indicate = obj.predict(wish, pray)

    return {
        'turn_symbol_desc': turn_symbol_desc,
        'main_indicate': main_indicate,
        'support_indicate': support_indicate
    }

if __name__ == '__main__':
    conf = {
        'wish': '恋爱',
        'pray': '找到对象',
    }
    results = handle(conf)
    print(results)