Update app.py

app.py

@@ -52,7 +52,7 @@ 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:
+    if img == None:
         img = cv2.imread('003e2c95d.jpg')
     #rgb_path = os.path.join(test_image_dir,img)
     #img = cv2.imread(rgb_path)
@@ -62,7 +62,7 @@ def gen_pred(img=inputs, model=seg_model):
     img = tf.expand_dims(img, axis=0)
     pred = model.predict(img)
     pred = np.squeeze(pred, axis=0)
-    return inputs, pred
+    return img, pred
 
 
 title = "<h1 style='text-align: center;'>Semantic Segmentation</h1>"