Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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/
|
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()
|