Update app.py

app.py

@@ -1,4 +1,7 @@
 import streamlit as st
+# wide layout
+st.set_page_config(layout="wide")
+
 from streamlit_drawable_canvas import st_canvas
 from PIL import Image
 from typing import Union
@@ -13,8 +16,6 @@ from config import HEIGHT, WIDTH, POS_PROMPT, NEG_PROMPT, COLOR_MAPPING, map_col
 from palette import COLOR_MAPPING_CATEGORY
 from preprocessing import preprocess_seg_mask, get_image, get_mask
 from explanation import make_inpainting_explanation, make_regeneration_explanation, make_segmentation_explanation
-# wide layout
-st.set_page_config(layout="wide")
 
 
 def on_upload() -> None: