Update app.py

app.py

@@ -144,9 +144,11 @@ 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:
-
+#with gr.Interface(css=css) as mainBody:
+gr.HTML("""<style>
+{css}
+</Style>""")
+  
 mainBody = gr.Interface(
     interface,
     [
@@ -160,7 +162,7 @@ mainBody = gr.Interface(
     [
         gr.components.Image(label="output")
     ],
-    css=css,
+    gr.html(css=css),
     #inputs="sketchpad",
     #outputs="label",
     theme="huggingface",