Uzaiir commited on
Commit
0ffc750
·
verified ·
1 Parent(s): 38dd61c

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +2 -0
src/streamlit_app.py CHANGED
@@ -2,6 +2,8 @@ import streamlit as st
2
  from io import BytesIO
3
  from PIL import Image
4
  # from utils import convert_to_bw, load_colorization_model, colorize_bw_image
 
 
5
 
6
  st.set_page_config(page_title="Image Colorizer", layout="centered")
7
 
 
2
  from io import BytesIO
3
  from PIL import Image
4
  # from utils import convert_to_bw, load_colorization_model, colorize_bw_image
5
+ import os
6
+ os.environ["STREAMLIT_SERVER_HEADLESS"] = "true"
7
 
8
  st.set_page_config(page_title="Image Colorizer", layout="centered")
9