Update app.py

app.py

@@ -58,11 +58,10 @@ image_output = gr.outputs.Image(type="pil", label="Output Image")
 
 
 def gen_pred(img=inputs, model=seg_model):
-    # rgb_path = os.path.join(test_image_dir,img)
-    # img = cv2.imread(rgb_path)
-    img = cv2.imread("./003e2c95d.jpg")
-    # pil_image = Image.open('./003b50a15.jpg')
-    # img = cv2.cvtColor(np.array(pil_image), cv2.COLOR_RGB2BGR)
+    pil_image = img.convert('RGB')
+    open_cv_image = numpy.array(pil_image)
+    img = open_cv_image[:, :, ::-1].copy() 
+    # img = cv2.imread("./003e2c95d.jpg")
     img = img[::IMG_SCALING[0], ::IMG_SCALING[1]]
     img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
     img = img/255