Update app.py

app.py

@@ -90,7 +90,7 @@ def draw_plot(pred_img, seg):
     ax.tick_params(width=0.0, labelsize=25)
     return fig
 
-def sepia():
+def sepia(input_img):
     input_img = Image.fromarray(input_img)
 
     inputs = feature_extractor(images=input_img, return_tensors="tf")