CrabInHoney
/

1D-CNN-MC-toxicity-classifier-ru

Text Classification

Model card Files Files and versions Community

CrabInHoney commited on 28 days ago

Commit

0b8b22b

•

1 Parent(s): 7d47775

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -54,7 +54,7 @@ Replacing ё with e.
         model_json = json_file.read()
     model = keras.models.model_from_json(model_json)
     # Загрузка весов из safetensors
-    state_dict = load_file(os.path.join(model_dir, 'model_weights.safetensors'))
     weights = [state_dict[f'weight_{i}'] for i in range(len(state_dict))]
     model.set_weights(weights)
     # Загрузка токенизатора

         model_json = json_file.read()
     model = keras.models.model_from_json(model_json)
     # Загрузка весов из safetensors
+    state_dict = load_file(os.path.join(model_dir, 'tf_model.safetensors'))
     weights = [state_dict[f'weight_{i}'] for i in range(len(state_dict))]
     model.set_weights(weights)
     # Загрузка токенизатора