Initial commit.

app.py

@@ -13,7 +13,7 @@ from torchvision import transforms
 # WEIGHTS = './weights/segmenter.pth
 WEIGHTS = './weights/segmenter_nusc.pth'
 FULL = True
-CACHE = False
+CACHE = True
 ALPHA = 0.5
 
 
@@ -233,7 +233,7 @@ iface = gr.Interface(predict, inputs=gr.Image(type='filepath'), title=title, des
                      # theme='dark',
                      outputs=[gr.Image(label="Pseudo segmentation", type="pil"),
                               gr.Image(label="Mapping to Cityscapes", type="pil")],
-                     examples=examples, cache_examples=False)
+                     examples=examples, cache_examples=CACHE)
 # iface = gr.Interface(predict, gr.inputs.Image(type='filepath'),
 #                      "image", title=title, description=description,
 #                      examples=examples)