Update main.py

main.py

@@ -76,10 +76,13 @@ css = '''
  '''
 gr.Interface(
     title=title,
-    
-    
+    description=description,
+    article=article,
     fn=predict_and_return_image,
-    inputs=[gr.Image(label="Gray Scale Image") , gr.Dropdown(["MODEL_1" , "MODEL_2"])],
+    inputs=[
+        gr.Image(label="Gray Scale Image"),
+        gr.Dropdown(["MODEL_1", "MODEL_2"])
+    ],
     outputs=[gr.Image(label="Predicted Colored Image")],
     css = '''
 .gradio-container {background-color: #DCDCDC}