Update main.py

main.py

@@ -81,4 +81,18 @@ gr.Interface(
     fn=predict_and_return_image,
     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: #228B22}
+.animate-spin {
+    animation: spin 1s linear infinite;
+}
+@keyframes spin {
+    from {
+        transform: rotate(0deg);
+    }
+    to {
+        transform: rotate(360deg);
+    }
+}
+ '''
 ).launch()