Upload examples/basic_usage.py with huggingface_hub

examples/basic_usage.py

@@ -15,7 +15,7 @@ def example_1_default_model():
     print("=" * 60)
     
     model = AutoModel.from_pretrained(
-        "pcdslab/dom-knn-models",
+        "SaeedLab/dom-formula-assignment-using-knn",
         trust_remote_code=True
     )
     
@@ -31,7 +31,7 @@ def example_2_variant_by_parameters():
     print("=" * 60)
     
     model = AutoModel.from_pretrained(
-        "pcdslab/dom-knn-models",
+        "SaeedLab/dom-formula-assignment-using-knn",
         data_source="7T-21T",
         k_neighbors=1,
         metric="euclidean",
@@ -54,7 +54,7 @@ def example_3_variant_by_name():
     print("=" * 60)
     
     model = AutoModel.from_pretrained(
-        "pcdslab/dom-knn-models",
+        "SaeedLab/dom-formula-assignment-using-knn",
         variant="knn_21T_k3_manhattan",
         trust_remote_code=True
     )
@@ -72,7 +72,7 @@ def example_4_make_predictions():
     
     # Load model
     model = AutoModel.from_pretrained(
-        "pcdslab/dom-knn-models",
+        "SaeedLab/dom-formula-assignment-using-knn",
         trust_remote_code=True
     )
     
@@ -117,7 +117,7 @@ def example_5_compare_models():
     
     for name, params in variants:
         model = AutoModel.from_pretrained(
-            "pcdslab/dom-knn-models",
+            "SaeedLab/dom-formula-assignment-using-knn",
             trust_remote_code=True,
             **params
         )
@@ -146,5 +146,5 @@ if __name__ == "__main__":
     except Exception as e:
         print(f"\n[ERROR] Error: {e}")
         print("\nNote: These examples require the model to be uploaded to")
-        print("      Hugging Face Hub at: pcdslab/dom-knn-models")
+        print("      Hugging Face Hub at: SaeedLab/dom-formula-assignment-using-knn")
         print("\nFor local testing, replace the repo ID with a local path.")