#安装包
import os
os.system('pip install -U openi')
#导入包
from openi.context import prepare, upload_openi

#初始化导入数据集和预训练模型到容器内
openi_context = prepare()

#获取数据集路径，预训练模型路径，输出路径
dataset_path = openi_context.dataset_path
pretrain_model_path = openi_context.pretrain_model_path
output_path = openi_context.output_path

print("dataset_path:")
os.listdir(dataset_path)

print("pretrain_model_path:")
os.listdir(pretrain_model_path)

print("output_path:")
os.listdir(output_path)

#回传结果到openi
upload_openi()