Update README.md

README.md

@@ -109,7 +109,7 @@ import joblib
 
 class CuisineClassifier:
 
-    def __init__(self, dataset=None):
+    def __init__(self, classifier="region"):
         print("Initializing CuisineClassifier...")
 
         components = ["cuisine_pipeline", "label_encoder"]
@@ -122,6 +122,8 @@ class CuisineClassifier:
                 paths[name] = hf_hub_download(
                     repo_id="NoahMeissner/CuisineClassifier", 
                     filename=f"region_classifier/{name}.joblib"
+                    if classifier == "cuisine":
+                      filename=f"region_classifier/{name}.joblib"
                 )
                 print(f"{name} downloaded.")
             except Exception as e: