Update app.py

app.py

@@ -215,7 +215,7 @@ def sepia(input_img):
 
     logits = np.transpose(logits, (0, 2, 3, 1))
     seg = np.argmax(logits, axis=-1)[0].astype('float32')
-    seg = cv2.resize(seg, (640, 640))
+    seg = cv2.resize(seg, (640, 640)).astype('uint8')
     
     color_seg = np.zeros(
         (seg.shape[0], seg.shape[1], 3), dtype=np.uint8