Spaces:

chansung
/

testtest

Runtime error

chansung commited on Jul 24, 2022

Commit

c90545e

1 Parent(s): 9cbcc8a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -206,7 +206,7 @@ def draw_plot(pred_img, seg):
     return fig
 def sepia(input_img):
-    img = cv2.imread(input_img)
     img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
     img_batch = np.expand_dims(img, axis=0)

     return fig
 def sepia(input_img):
+    img = cv2.imread(input_img).astype(np.float32)
     img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
     img_batch = np.expand_dims(img, axis=0)