fix

core/model.py

@@ -21,16 +21,15 @@ def build_classifier():
   output = keras.layers.Dense(num_classes)
   return keras.Sequential(layers=[model, output])
 
-def load_classifier():
+def load_classifier(classifier_path):
   backbone = load_backbone()
   model = movinet_model.MovinetClassifier(
     backbone=backbone,
-    num_classes=num_classes,
-    output_states=True)
+    num_classes=600)
   model.build([batch_size, num_frames, resolution, resolution, 3])
   output = keras.layers.Dense(num_classes)
   model = keras.Sequential(layers=[model, output])
-  model.load_weights(model_save_path)
+  model.load_weights(classifier_path)
   return model
 
 def compile_classifier(model):