badru commited on
Commit
b930f66
·
verified ·
1 Parent(s): 0d536da

Create mmsocrapp.py

Browse files
Files changed (1) hide show
  1. mmsocrapp.py +24 -0
mmsocrapp.py ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from transformers import TrOCRProcessor, VisionEncoderDecoderModel
3
+ from PIL import Image
4
+
5
+ # Load Hugging Face model
6
+ processor = TrOCRProcessor.from_pretrained("microsoft/trocr-base-handwritten")
7
+ model = VisionEncoderDecoderModel.from_pretrained("microsoft/trocr-base-handwritten")
8
+
9
+ st.title("OCR Application")
10
+ uploaded_file = st.file_uploader("Upload an image", type=["png", "jpg", "jpeg"])
11
+
12
+ if uploaded_file:
13
+ # Process the image
14
+ image = Image.open(uploaded_file).convert("RGB")
15
+ st.image(image, caption="Uploaded Image", use_column_width=True)
16
+
17
+ # Perform OCR
18
+ pixel_values = processor(images=image, return_tensors="pt").pixel_values
19
+ generated_ids = model.generate(pixel_values)
20
+ generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
21
+
22
+ # Display OCR result
23
+ st.text("Extracted Text:")
24
+ st.write(generated_text)