Update README.md

README.md

@@ -42,15 +42,15 @@ from transformers import VisionEncoderDecoderModel
 from transformers.models.nougat import NougatTokenizerFast
 from nougat_latex import NougatLaTexProcessor
 
-
+model_name = "Norm/nougat-latex-base"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 # init model
-model = VisionEncoderDecoderModel.from_pretrained("Norm/nougat-latex-base").to(device)
+model = VisionEncoderDecoderModel.from_pretrained(model_name).to(device)
 
 # init processor
-tokenizer = NougatTokenizerFast.from_pretrained("Norm/nougat-latex-base")
+tokenizer = NougatTokenizerFast.from_pretrained(model_name)
 
-latex_processor = NougatLaTexProcessor.from_pretrained("Norm/nougat-latex-base")
+latex_processor = NougatLaTexProcessor.from_pretrained(model_name)
 
 # run test
 image = Image.open("path/to/latex/image.png")