Update app.py

app.py

@@ -24,7 +24,16 @@ init_session_states()
 import easyocr
 from transformers import TrOCRProcessor, VisionEncoderDecoderModel
 
+def text_recognition(image):
+    processor = TrOCRProcessor.from_pretrained("microsoft/trocr-base-handwritten") 
+    model = VisionEncoderDecoderModel.from_pretrained("microsoft/trocr-base-handwritten")
+    #processor = TrOCRProcessor.from_pretrained("microsoft/trocr-large-handwritten") 
+    #model = VisionEncoderDecoderModel.from_pretrained("microsoft/trocr-large-handwritten")
 
+    pixel_values = processor(image, return_tensors="pt").pixel_values 
+    generated_ids = model.generate(pixel_values)
+    generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0] 
+    st.write(generated_text)
 
 def main():
    
@@ -35,14 +44,12 @@ def main():
         st.write(file_details)
         image = load_image(uploaded_file)
         st.image(image,width=500)
-        #st.session_state['disp'].text("Detecting text....")
-        st.write("Detecting text...")
+        st.write("Detecting text bounding box...")
         reader = easyocr.Reader(['en'],gpu=True)
         bound = reader.readtext(image)
-        st.write("Detection complete")
+        st.write("Bounding box Detection complete")
         st.write(str(bound))
-        #st.session_state['disp'].text("Detection complete")
-
+        text_recognition(image)