From 17f5f4be923bda379124cb09b0b5692396d504fc Mon Sep 17 00:00:00 2001
From: zhaoting <zhaoting23@huawei.com>
Date: Wed, 7 Apr 2021 10:43:55 +0800
Subject: [PATCH] add requirements for unet

---
 model_zoo/official/cv/unet/README.md        | 1 +
 model_zoo/official/cv/unet/README_CN.md     | 1 +
 model_zoo/official/cv/unet/requirements.txt | 2 ++
 model_zoo/official/nlp/tinybert/export.py   | 2 +-
 4 files changed, 5 insertions(+), 1 deletion(-)
 create mode 100644 model_zoo/official/cv/unet/requirements.txt

diff --git a/model_zoo/official/cv/unet/README.md b/model_zoo/official/cv/unet/README.md
index 9d413e7f4a..c902274a34 100644
--- a/model_zoo/official/cv/unet/README.md
+++ b/model_zoo/official/cv/unet/README.md
@@ -144,6 +144,7 @@ Then you can run everything just like on ascend.
         ├── mindspore_hub_conf.py           // hub config file
         ├── postprocess.py                  // unet 310 infer postprocess.
         ├── preprocess.py                   // unet 310 infer preprocess dataset
+        ├── requirements.txt                // Requirements of third party package.
 ```
 
 ### [Script Parameters](#contents)
diff --git a/model_zoo/official/cv/unet/README_CN.md b/model_zoo/official/cv/unet/README_CN.md
index e40d3d3f86..db67c45bab 100644
--- a/model_zoo/official/cv/unet/README_CN.md
+++ b/model_zoo/official/cv/unet/README_CN.md
@@ -148,6 +148,7 @@ bash scripts/docker_start.sh unet:20.1.0 [DATA_DIR] [MODEL_DIR]
         ├── mindspore_hub_conf.py           // hub 配置脚本
         ├── postprocess.py                  // 310 推理后处理脚本
         ├── preprocess.py                   // 310 推理前处理脚本
+        ├── requirements.txt                // 需要的三方库.
 ```
 
 ### 脚本参数
diff --git a/model_zoo/official/cv/unet/requirements.txt b/model_zoo/official/cv/unet/requirements.txt
new file mode 100644
index 0000000000..542ad16815
--- /dev/null
+++ b/model_zoo/official/cv/unet/requirements.txt
@@ -0,0 +1,2 @@
+opencv-python >= 4.1.2.30
+pillow >= 6.2.0
diff --git a/model_zoo/official/nlp/tinybert/export.py b/model_zoo/official/nlp/tinybert/export.py
index 61847600d6..65f6edbfce 100644
--- a/model_zoo/official/nlp/tinybert/export.py
+++ b/model_zoo/official/nlp/tinybert/export.py
@@ -48,7 +48,7 @@ task_params = {"SST-2": {"num_labels": 2, "seq_length": 64},
                "QNLI": {"num_labels": 2, "seq_length": 128},
                "MNLI": {"num_labels": 3, "seq_length": 128},
                "TNEWS": {"num_labels": 15, "seq_length": 128},
-               "CLUENER": {"num_labels": 10, "seq_length": 128}}
+               "CLUENER": {"num_labels": 43, "seq_length": 128}}
 
 class Task:
     """