[fix] input to inputs

This commit is contained in:
Nat Lee 2024-12-17 14:02:30 +08:00
parent 38c06522a5
commit b9418eb46f

View File

@ -97,7 +97,7 @@ def load_model(
# -------------------------- # --------------------------
race_model = Model(inputs=model.input, outputs=base_model_output) race_model = Model(inputs=model.inputs, outputs=base_model_output)
# -------------------------- # --------------------------