Amending retrain-pipeline as per Model-deployment requirements

1. Model-path is deduced by modelversion
2. Only retrain-model will be present in Model.zip

Change-Id: I0bc49b876b1d298b8d558e48f4c32c49fde2f0bf
Signed-off-by: ashishj1729 <jain.ashish@samsung.com>

diff --git a/kf-pipelines/qoe-pipeline-retrain-2.ipynb b/kf-pipelines/qoe-pipeline-retrain-2.ipynb
index f933c35..f0ab77d 100644 (file)
--- a/kf-pipelines/qoe-pipeline-retrain-2.ipynb
+++ b/kf-pipelines/qoe-pipeline-retrain-2.ipynb
@@ -131,10 +131,7 @@
     "\n",
     "    # Load the model in SavedModel format\n",
     "    # Path to the directory containing the saved model\n",
-    "    if artifactversion ==\"1.0.0\":\n",
-    "        model_path = \"./Model/1\"\n",
-    "    else:\n",
-    "        model_path= \"./Model/1/retrain\"\n",
+    "    model_path = f\"./Model/{modelversion}\"\n",
     "        \n",
     "    model = tf.keras.models.load_model(model_path)\n",
     "    model.compile(loss='mse', optimizer='adam', metrics=['mse'])\n",
@@ -199,7 +196,7 @@
     "    print(updated_model_info)\n",
     "    \n",
     "    mm_sdk.upload_metrics(data, modelname, modelversion,new_artifactversion)\n",
-    "    mm_sdk.upload_model(\"./\", modelname, modelversion, new_artifactversion)\n"
+    "    mm_sdk.upload_model(\"./retrain/\", modelname, modelversion, new_artifactversion)\n"
    ]
   },
   {