Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 
 import numpy as np
-from PIL import Image
+import cv2
 from matplotlib import gridspec
 import matplotlib.pyplot as plt
 import onnxruntime as ort
@@ -206,9 +206,9 @@ def draw_plot(pred_img, seg):
     return fig
 
 def sepia(input_img):
-    img = Image.fromarray(input_img)
+    img = cv2.imread(input_img)
+    img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
     img_batch = np.expand_dims(img, axis=0)
-    img_batch = img_batch.astype(float)
     
     outputs = sess.run(None, {"pixel_values": img_batch})[0]    
     logits = outputs.logits