MaxLSB commited on
Commit
892e21c
·
verified ·
1 Parent(s): a7a20a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ MEDIA_PATH = "media/le-carnet.png" # Relative path to logo
16
  models = {}
17
  tokenizers = {}
18
  for name in MODEL_NAMES:
19
- hub_id = f"MaxLSB/LeCarnet-{name.split('-')[-1]}M"
20
  tokenizers[name] = AutoTokenizer.from_pretrained(hub_id, token=HF_TOKEN)
21
  models[name] = AutoModelForCausalLM.from_pretrained(hub_id, token=HF_TOKEN)
22
  models[name].eval()
 
16
  models = {}
17
  tokenizers = {}
18
  for name in MODEL_NAMES:
19
+ hub_id = f"MaxLSB/{name}"
20
  tokenizers[name] = AutoTokenizer.from_pretrained(hub_id, token=HF_TOKEN)
21
  models[name] = AutoModelForCausalLM.from_pretrained(hub_id, token=HF_TOKEN)
22
  models[name].eval()