hugbump commited on
Commit
ab27eac
1 Parent(s): a46ae31

Input Image has to be greyscale, otherwise broadcast error. Thus add .convert('L')

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -19,7 +19,8 @@ def process_image(image):
19
  st.title('Streamlit Replication of nielsr/TrOCR-handwritten')
20
  uploaded_file = st.file_uploader("Choose an image...")
21
  if uploaded_file:
22
- input_image = Image.open(uploaded_file)
 
23
  st.image(uploaded_file, caption='Input Image', use_column_width=True)
24
  generated_text = process_image(input_image)
25
  st.write(generated_text)
 
19
  st.title('Streamlit Replication of nielsr/TrOCR-handwritten')
20
  uploaded_file = st.file_uploader("Choose an image...")
21
  if uploaded_file:
22
+ # .convert('L') to greyscale
23
+ input_image = Image.open(uploaded_file).convert('L')
24
  st.image(uploaded_file, caption='Input Image', use_column_width=True)
25
  generated_text = process_image(input_image)
26
  st.write(generated_text)