Update app.py

app.py

@@ -52,10 +52,11 @@ inputs = gr.inputs.Image(type="pil", label="Upload an image")
 # outputs = gr.outputs.HTML() #uncomment for single class output 
 
 def gen_pred(img=inputs, model=seg_model):
-    if img == None:
-        img = cv2.imread('./003e2c95d.jpg')
+
+    img = cv2.imread('./003e2c95d.jpg')
     path = os.getcwd()
-    os.listdir(path)
+    tmp = os.listdir(path)
+    print(tmp)
     #rgb_path = os.path.join(test_image_dir,img)
     #img = cv2.imread(rgb_path)
     img = img[::IMG_SCALING[0], ::IMG_SCALING[1]]