HuggingFaceFW
/

fineweb-edu-classifier

Text Classification

Inference Endpoints

Model card Files Files and versions Community

anton-l HF staff commited on Jun 3

Commit

d7b750e

•

1 Parent(s): 172e214

Update src/run_edu_bert.py

Files changed (1) hide show

src/run_edu_bert.py +1 -1

src/run_edu_bert.py CHANGED Viewed

@@ -39,7 +39,7 @@ def main(args):
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
-    parser.add_argument("--model_name", type=str, default="HHuggingFaceFW/fineweb-edu-classifier")
     parser.add_argument("--dataset_name", type=str, default="HuggingFaceFW/fineweb")
     parser.add_argument("--dataset_config", type=str, default="default")
     parser.add_argument("--output_dataset_name", type=str, default="HuggingFaceFW/fineweb-edu")

 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
+    parser.add_argument("--model_name", type=str, default="HuggingFaceFW/fineweb-edu-classifier")
     parser.add_argument("--dataset_name", type=str, default="HuggingFaceFW/fineweb")
     parser.add_argument("--dataset_config", type=str, default="default")
     parser.add_argument("--output_dataset_name", type=str, default="HuggingFaceFW/fineweb-edu")