Update README.md

README.md

@@ -39,18 +39,14 @@ Extensible Configuration: Leverages a custom configuration setup that can be eas
 
 **How to load and call Kraken-Multilingual model :**
 ```
-from transformers import AutoConfig, AutoModelForCausalLM
-from configuration_kraken import KrakenConfig
-from modeling_kraken import KrakenForCausalLM
+from transformers import AutoModelForCausalLM
+
 
-AutoConfig.register("kraken", KrakenConfig)
-AutoModelForCausalLM.register(KrakenConfig, KrakenForCausalLM)
 
 device = "cuda:0" ## Setup "cuda:0" if NVIDIA, "mps" if on Mac
 
 # Load the model and config:
-config = AutoConfig.from_pretrained("./kraken_model")
-model = AutoModelForCausalLM.from_pretrained("./kraken_model", config=config, trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("./kraken_model", trust_remote_code=True)
 ```
 
 # Call the German expert: