Update load_model.py

load_model.py

@@ -1,6 +1,6 @@
-from transformers import AutoModelForCausalLM, AutoTokenizer
-
-
-model_checkpoint = "as-cle-bert/saccharomyces-pythia-v1"
-model = AutoModelForCausalLM.from_pretrained(model_checkpoint)
+from transformers import AutoModelForCausalLM, AutoTokenizer
+
+
+model_checkpoint = "as-cle-bert/tiny-fungal-llama"
+model = AutoModelForCausalLM.from_pretrained(model_checkpoint)
 tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)