20f3472 7c5369d 20f3472
1
2
3
4
5
6
7
8
from transformers import AutoModelForCausalLM, AutoTokenizer def load_model(): model_name = "Erfan11/Neuracraft" tokenizer = AutoTokenizer.from_pretrained(model_name) model = AutoModelForCausalLM.from_pretrained(model_name) return model, tokenizer
