Update app.py

app.py

@@ -1,4 +1,11 @@
 import gradio as gr
+import opencv as cv2
+
+def color_to_gray(img):
+    new_img = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
+    return new_img
+
+
 
 def predict(video_in, image_in_video, image_in_img):
     if video_in == None and image_in_video == None and image_in_img == None:
@@ -48,6 +55,7 @@ with gr.Blocks() as blocks:
                                          outputs=[image_in_video, image_in_img], queue=False, show_progress=False)
             with gr.Column():
                 image_out = gr.Image()
+                image_out = color_to_gray(image_out)
         run_btn = gr.Button("Run")
         run_btn.click(fn=predict, inputs=[
                       image_in_img, image_in_video], outputs=[image_out])