hugbump commited on
Commit
4e9737c
1 Parent(s): ab27eac

Should .convert('RGB') instead of .convert('L')

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