Update app.py
Browse files
app.py
CHANGED
|
@@ -7,6 +7,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
|
|
| 7 |
|
| 8 |
# scratch with latbert tokenizer
|
| 9 |
CHECKPOINT_PATH= 'scratch_2-nodes_tokenizer_latbert-original_packing_fcocchi/model.safetensors'
|
|
|
|
| 10 |
|
| 11 |
print(f"Loading model from: {CHECKPOINT_PATH}")
|
| 12 |
tokenizer = AutoTokenizer.from_pretrained(CHECKPOINT_PATH)
|
|
|
|
| 7 |
|
| 8 |
# scratch with latbert tokenizer
|
| 9 |
CHECKPOINT_PATH= 'scratch_2-nodes_tokenizer_latbert-original_packing_fcocchi/model.safetensors'
|
| 10 |
+
CHECKPOINT_PATH= 'itserr/latin_llm_alpha'
|
| 11 |
|
| 12 |
print(f"Loading model from: {CHECKPOINT_PATH}")
|
| 13 |
tokenizer = AutoTokenizer.from_pretrained(CHECKPOINT_PATH)
|