Update app.py

app.py

@@ -16,10 +16,12 @@ if file is not None:
     processor = TrOCRProcessor.from_pretrained('microsoft/trocr-small-handwritten')
     model = VisionEncoderDecoderModel.from_pretrained('microsoft/trocr-small-handwritten')
     pixel_values = processor(images=image, return_tensors="pt").pixel_values
+    generated_ids = model.generate(pixel_values)
+    generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
     
     # Display the image
     st.image(image, caption='Uploaded Image', use_column_width=True)
     
     # Print the shape of the image
-    st.write(f"Image shape: {image.size}")
+    st.write(f"Text: {generated_text}")