shenzye46 commited on
Commit
9530fd9
·
verified ·
1 Parent(s): 8dda031

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,8 +6,8 @@ model_name = "shenzye46/SmolLM-135M-fine-tuned-recepie" # Replace with your mod
6
  tokenizer_name = "HuggingFaceTB/SmolLM-135M"
7
 
8
  def load_model_and_tokenizer():
9
- tokenizer = AutoTokenizer.from_pretrained(model_name)
10
- model = AutoModelForCausalLM.from_pretrained(tokenizer_name)
11
  return tokenizer, model
12
 
13
  tokenizer, model = load_model_and_tokenizer()
 
6
  tokenizer_name = "HuggingFaceTB/SmolLM-135M"
7
 
8
  def load_model_and_tokenizer():
9
+ tokenizer = AutoTokenizer.from_pretrained(tokenizer_name)
10
+ model = AutoModelForCausalLM.from_pretrained(model_name)
11
  return tokenizer, model
12
 
13
  tokenizer, model = load_model_and_tokenizer()