Spaces:

Heramb26
/

TR-OCR-CustomModel

Sleeping

Heramb26 commited on Nov 22, 2024

Commit

2f2995e

1 Parent(s): 0218a3d

Chnages to APP

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ checkpoint_folder = hf_hub_download(repo_id="Heramb26/tr-ocr-custom-checkpoints"
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 # Load the fine-tuned model and processor from the downloaded folder
-model = VisionEncoderDecoderModel.from_pretrained(checkpoint_folder).to(device)
 processor = TrOCRProcessor.from_pretrained("microsoft/trocr-large-handwritten")
 def ocr_image(image):

 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 # Load the fine-tuned model and processor from the downloaded folder
+model = VisionEncoderDecoderModel.from_pretrained("Heramb26/tr-ocr-custom-checkpoints/checkpoint-2070").to(device)
 processor = TrOCRProcessor.from_pretrained("microsoft/trocr-large-handwritten")
 def ocr_image(image):