added retrain code

2025-11-11 20:38:13 +01:00
parent df6056fb86
commit 95d16690ef
2 changed files with 55 additions and 37 deletions
--- a/epdb/views.py
+++ b/epdb/views.py
@ -955,6 +955,12 @@ def package_model(request, package_uuid, model_uuid):
                ]
                dispatch(current_user, evaluate_model, current_model.pk, multigen, eval_package_ids)

+                return redirect(current_model.url)
+            elif hidden == "retrain":
+                from .tasks import dispatch, retrain
+
+                dispatch(current_user, retrain, current_model.pk)
+
                return redirect(current_model.url)
            else:
                return HttpResponseBadRequest()