DeepDiveDev commited on
Commit
6477a5b
·
verified ·
1 Parent(s): 224f1ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -5,7 +5,12 @@ from PIL import Image
5
  from transformers import AutoProcessor, AutoModelForVision2Seq
6
 
7
  # Load the model & processor
8
- model_name = "Murasajo/Llama-3.2-VL-Finetuned-on-HandwrittenText"
 
 
 
 
 
9
  processor = AutoProcessor.from_pretrained(model_name)
10
  model = AutoModelForVision2Seq.from_pretrained(model_name)
11
 
 
5
  from transformers import AutoProcessor, AutoModelForVision2Seq
6
 
7
  # Load the model & processor
8
+ from transformers import TrOCRProcessor, VisionEncoderDecoderModel
9
+
10
+ model_name = "microsoft/trocr-large-handwritten"
11
+ processor = TrOCRProcessor.from_pretrained(model_name)
12
+ model = VisionEncoderDecoderModel.from_pretrained(model_name)
13
+
14
  processor = AutoProcessor.from_pretrained(model_name)
15
  model = AutoModelForVision2Seq.from_pretrained(model_name)
16