Update README.md

README.md

@@ -168,7 +168,12 @@ class IAMDataset(Dataset):
 
 # Load the dataset
 from transformers import TrOCRProcessor, AutoTokenizer
-processor = TrOCRProcessor.from_pretrained("magistermilitum/tridis_v2_HTR_historical_manuscripts") #change to this model
+
+# Download this model from Hugging Face Hub
+model_path = hf_hub_download(repo_id="magistermilitum/tridis_v2_HTR_historical_manuscripts")
+
+#Load the processor from the model
+processor = TrOCRProcessor.from_pretrained(model_path)
 
 # Define the dataset
 #Open the file with text lines
@@ -209,7 +214,8 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 from safetensors.torch import load_file
 # Load the weights of this model
-state_dict= load_file("tridis_v2_HTR_historical_manuscripts/model.safetensors")
+safetensors_path = f"{model_path}/model.safetensors" 
+state_dict = load_file(safetensors_path)
 
 # Load the trocr model
 model = VisionEncoderDecoderModel.from_pretrained("microsoft/trocr-large-handwritten")