Spaces:

Heramb26
/

TR-OCR-CustomModel

Sleeping

Heramb26 commited on Nov 22, 2024

Commit

e705ba3

1 Parent(s): bbe7f9b

Chnages

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ import os
 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").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/TC-OCR-Custom").to(device)
 processor = TrOCRProcessor.from_pretrained("microsoft/trocr-large-handwritten")
 def ocr_image(image):