Update app.py

app.py

@@ -6,7 +6,6 @@ def label_func(fn): return path/'masks1b-binary'/f'{fn.stem}.png'
 
 repo_id = "hugginglearners/kvasir-seg"
 learn = from_pretrained_fastai(repo_id)
-#labels = learn.dls.vocab
 
 def predict(img):
     img = PILImage.create(img)
@@ -16,7 +15,6 @@ def predict(img):
 interface_options = {
     "title": "kvasir-seg fastai segmentation",
     "description": "tbd",
-    "interpretation": "default",
     "layout": "horizontal",
     "examples": [
         "cju5eftctcdbj08712gdp989f.jpg",
@@ -28,8 +26,8 @@ interface_options = {
 
 demo = gr.Interface(
     fn=predict,
-    inputs=gr.inputs.Image(shape=(224, 224)),
-    outputs=gr.inputs.Image(shape=(224, 224)),
+    inputs=gr.Image(shape=(224, 224)),
+    outputs=gr.Image(shape=(224, 224)),
     **interface_options,
 )