CocoRoF commited on
Commit
eaa7e57
·
verified ·
1 Parent(s): b6bc975

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -23,12 +23,12 @@ It achieves the following results on the evaluation set:
23
  repo_id = "x2bee/ModernBERT-ecs-GIST-category"
24
 
25
  model_file = hf_hub_download(repo_id=repo_id, filename="model.py", token=token)
26
- spec = importlib.util.spec_from_file_location("PlateerCategoryClassifier", model_file)
27
  module = importlib.util.module_from_spec(spec)
28
  spec.loader.exec_module(module)
29
- PlateerCategoryClassifier = module.PlateerCategoryClassifier
30
 
31
- clf_model = PlateerCategoryClassifier(repo_id)
32
 
33
  # use
34
  text = '반팔티셔츠'
 
23
  repo_id = "x2bee/ModernBERT-ecs-GIST-category"
24
 
25
  model_file = hf_hub_download(repo_id=repo_id, filename="model.py", token=token)
26
+ spec = importlib.util.spec_from_file_location("MajorClassifier", model_file)
27
  module = importlib.util.module_from_spec(spec)
28
  spec.loader.exec_module(module)
29
+ MajorClassifier = module.MajorClassifier
30
 
31
+ clf_model = MajorClassifier(repo_id)
32
 
33
  # use
34
  text = '반팔티셔츠'