Upload modeling_knn.py with huggingface_hub

modeling_knn.py

@@ -96,11 +96,11 @@ class KNNModel(PreTrainedModel):
 
         Examples:
             # Load default best model
-            model = KNNModel.from_pretrained("pcdslab/dom-knn-models")
+            model = KNNModel.from_pretrained("SaeedLab/dom-formula-assignment-using-knn")
 
             # Load specific variant by parameters
             model = KNNModel.from_pretrained(
-                "pcdslab/dom-knn-models",
+                "SaeedLab/dom-formula-assignment-using-knn",
                 data_source="7T-21T",  # This is an ensemble!
                 k_neighbors=1,
                 metric="euclidean"
@@ -108,7 +108,7 @@ class KNNModel(PreTrainedModel):
 
             # Load by variant name
             model = KNNModel.from_pretrained(
-                "pcdslab/dom-knn-models",
+                "SaeedLab/dom-formula-assignment-using-knn",
                 variant="knn_21T_k3_manhattan"
             )
         """