DeepDiveDev commited on
Commit
30abd6a
·
verified ·
1 Parent(s): fa36a00

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -9,8 +9,9 @@ processor1 = TrOCRProcessor.from_pretrained("DeepDiveDev/transformodocs-ocr")
9
  model1 = VisionEncoderDecoderModel.from_pretrained("DeepDiveDev/transformodocs-ocr")
10
 
11
  # Load the fallback model (allenai/olmOCR-7B-0225-preview)
12
- processor2 = AutoProcessor.from_pretrained("allenai/olmOCR-7B-0225-preview")
13
- model2 = AutoModelForVision2Seq.from_pretrained("allenai/olmOCR-7B-0225-preview", torch_dtype=torch.float16)
 
14
 
15
  # Function to extract text using both models
16
  def extract_text(image):
 
9
  model1 = VisionEncoderDecoderModel.from_pretrained("DeepDiveDev/transformodocs-ocr")
10
 
11
  # Load the fallback model (allenai/olmOCR-7B-0225-preview)
12
+ model2 = VisionEncoderDecoderModel.from_pretrained("microsoft/trocr-base-handwritten")
13
+ processor2 = TrOCRProcessor.from_pretrained("microsoft/trocr-base-handwritten")
14
+
15
 
16
  # Function to extract text using both models
17
  def extract_text(image):