Update app.py

app.py

@@ -20,6 +20,24 @@ def interface(image, model: str = "eccv16"):
     )
     return output_img
 
+css='''
+.Box {
+	background-color: var(--color-canvas-default);
+	border-color: var(--color-border-default);
+	border-style: solid;
+	border-width: 1px;
+	border-radius: 6px;
+}
+.d-flex {
+	display: flex !important;
+}
+.flex-md-row {
+	flex-direction: row !important;
+}
+.flex-column {
+	flex-direction: column !important;
+}
+'''
 title = "Image Colorization Using AI Models"
 description = r"""<center>An automatic colorization functionality for Real-Time User-Guided Image Colorization with Learned Deep Priors,ECCV16 & SIGGRAPH 2017 Models!<br>
 Practically the algorithm is used to COLORIZE your **old BLACK & WHITE / GRAYSCALE photos**.<br>
@@ -126,6 +144,9 @@ Given a grayscale photograph as input, this demo attacks the problem of hallucin
 <center><img src='https://visitor-badge.glitch.me/badge?page_id=dj_colorization_eccv16siggraph17' alt='visitor badge'></center>
 """
 
+mainBody = gr.Blocks(css=css)
+#with gr.Blocks(css=".gradio-container {background-color: red}") as demo:
+
 mainBody = gr.Interface(
     interface,
     [
@@ -139,6 +160,7 @@ mainBody = gr.Interface(
     [
         gr.components.Image(label="output")
     ],
+    css=css,
     #inputs="sketchpad",
     #outputs="label",
     theme="huggingface",