Update app.py

app.py

@@ -5,16 +5,10 @@ import os
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 
-# finetuning with original tokenizer
-CHECKPOINT_PATH= '/work/pnrr_itserr/latin_model/cineca_checkpoints/finetuning-gpt2_2-nodes_tokenizer-original_lr-6e-4_fcocchi'
-# finetuning with our embedding and tokenizer
-##CHECKPOINT_PATH= '/work/pnrr_itserr/latin_model/cineca_checkpoints/finetuning-gpt2_2-nodes_tokenizer-our-embedding_lr-6e-4_fcocchi'
-# scratch with original tokenizer
-CHECKPOINT_PATH= '/work/pnrr_itserr/latin_model/cineca_checkpoints/scratch_2-nodes_tokenizer-original_packing_fcocchi'
 # scratch with latbert tokenizer
-CHECKPOINT_PATH= '/work/pnrr_itserr/latin_model/cineca_checkpoints/scratch_2-nodes_tokenizer_latbert-original_packing_fcocchi'
-print(f"Loading model from: {CHECKPOINT_PATH}")
+CHECKPOINT_PATH= 'scratch_2-nodes_tokenizer_latbert-original_packing_fcocchi/model.safetensors'
 
+print(f"Loading model from: {CHECKPOINT_PATH}")
 tokenizer = AutoTokenizer.from_pretrained(CHECKPOINT_PATH)
 model = AutoModelForCausalLM.from_pretrained(CHECKPOINT_PATH)