train is ready

manual_run reuses train's export
add current_model to constants

manual_run.py

@@ -2,7 +2,8 @@ import shutil
 from datetime import datetime
 from os import path
 
-from constants import models_dir, prefix, date_format, MODELS_FOLDER
+import train
+from constants import models_dir, prefix, date_format, MODELS_FOLDER, CURRENT_ONNX_MODEL
 from api_client import ApiClient
 from augmentation import Augmentator
 from exports import export_onnx
@@ -20,12 +21,5 @@ shutil.copytree(result_dir, model_dir, dirs_exist_ok=True)
 model_path = path.join(models_dir, f'{prefix[:-1]}.pt')
 shutil.copy(path.join(model_dir, 'weights', 'best.pt'), model_path)
 
-api_client = ApiClient()
-onnx_path = export_onnx(model_path)
-print(f'Conversion done: onnx path: {onnx_path}')
-
-with open(onnx_path, 'rb') as binary_file:
-    onnx_bytes = binary_file.read()
-
-key = Security.get_model_encryption_key()
-api_client.upload_big_small_resource(onnx_bytes, onnx_path, MODELS_FOLDER, key)
+train.export_current_model()
+print('success!')