From 3e38b572ce65c5245e5bd3134cd619bd4867ae8f Mon Sep 17 00:00:00 2001 From: Michael Duong Date: Tue, 4 Nov 2025 16:52:24 +0000 Subject: [PATCH] remove the specific fast ai prediction --- deployment/handlers/prediction_app.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/deployment/handlers/prediction_app.py b/deployment/handlers/prediction_app.py index e8b02ac..f1036d3 100644 --- a/deployment/handlers/prediction_app.py +++ b/deployment/handlers/prediction_app.py @@ -64,11 +64,11 @@ def warming_up_invocation( columns=model.model.original_features, ) - model_names = model.model.model_names() - if "NeuralNetFastAI" in model_names: - model.model.predict(warmup_df, model="NeuralNetFastAI") - else: - model.predict(data=warmup_df) + # model_names = model.model.model_names() + # if "NeuralNetFastAI" in model_names: + # model.model.predict(warmup_df, model="NeuralNetFastAI") + # else: + model.predict(data=warmup_df) def handler(event, context):