diff --git a/epdb/views.py b/epdb/views.py index 9af2ebe5..784287ca 100644 --- 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() diff --git a/templates/modals/objects/retrain_model_modal.html b/templates/modals/objects/retrain_model_modal.html index 9ed7745f..8bc90301 100644 --- a/templates/modals/objects/retrain_model_modal.html +++ b/templates/modals/objects/retrain_model_modal.html @@ -1,43 +1,55 @@ -