Update app.py

app.py

@@ -2,15 +2,16 @@ import gradio as gr
 import numpy as np
 
 def sepia(input_img):
-    sepia_filter = np.array([[0.393,0.769,0.189],[0.349,0.686,0.168],[0.272,0.534,0.131]])
+    sepia_filter = np.array([[0.393, 0.769, 0.189], [0.349, 0.686, 0.168], [0.272, 0.534, 0.131]])
     sepia_img = input_img.dot(sepia_filter.T)
     sepia_img /= sepia_img.max()
-    print(input_img.shape,sepia_img.shape)
+    print(input_img.shape, sepia_img.shape)
     return sepia_img
 
 demo = gr.Interface(
-    sepia,  
-    gr.Image(shape=(200,200)),"image"
+    sepia,
+    gr.Image(element="image", image_mode="RGB"),
+    "image"
 )
 
 demo.launch()