Now include json in model zip along with onnx file and .pt file

scripts/train.sh

@@ -49,7 +49,7 @@ if [ ! -f "${UNZIPPED}/data.yaml" ]; then
 fi
 
 FULLDEST=$(realpath "${UNZIPPED}")
-python scripts/data.py "${FULLDEST}/data.yaml"
+python scripts/data.py "${FULLDEST}/data.yaml" "${NAME}"
 
 cd yolov5
 source env/bin/activate
@@ -84,11 +84,13 @@ fi
 
 MODELPATH=$(realpath "../models/${ID}/weights/best.onnx")
 PTPATH=$(realpath "../models/${ID}/weights/best.pt")
+JSONPATH=$(realpath "${FULLDEST}/${NAME}.json")
 ZIPPATH="${ID}.zip"
 
 mkdir zip
 cp "${MODELPATH}" "zip/${NAME}.onnx"
 cp "${PTPATH}" "zip/${NAME}.pt"
+cp "${JSONPATH}" "zip/${NAME}.json"
 cd zip
 zip "../${ZIPPATH}" *
 cd ..