niclasfw commited on
Commit
c66873e
1 Parent(s): 703c573

Adding code to download baseline model.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ from transformers import pipeline
21
 
22
  # tokenizer, model = get_model()
23
 
24
- model_id = "niclasfw/bloom-1b5-schlager-bot-004"
25
 
26
  model = AutoModelForCausalLM.from_pretrained(model_id)
27
  tokenizer = AutoTokenizer.from_pretrained(model_id)
 
21
 
22
  # tokenizer, model = get_model()
23
 
24
+ model_id = "niclasfw/schlager-bot-004"
25
 
26
  model = AutoModelForCausalLM.from_pretrained(model_id)
27
  tokenizer = AutoTokenizer.from_pretrained(model_id)