test class names

app.py

@@ -68,9 +68,10 @@ def treat_grayscale(img):
 
 def get_name_by_id(contiguous_id, ade20k_meta):
     stuff_classes = ade20k_meta["stuff_classes"]
+    ids = ade20k_meta["stuff_dataset_id_to_contiguous_id"]
 
     if contiguous_id < len(stuff_classes):
-        return stuff_classes[contiguous_id]
+        return stuff_classes[ids[contiguous_id]]
 
     return "Unknown"
 
@@ -115,7 +116,7 @@ gui = gr.Interface(
     predict,
     inputs=[
         gr.Image(type='filepath', label="Input Image"),
-        gr.Textbox(type='text', label="Input Image (URL)"),
+        gr.Textbox(type='text', label="Input Image (URL) - not considered if image was uploaded"),
         gr.Slider(minimum=0.0, maximum=1.0, step=0.01, value=0.05, label="Confidence Threshold")
     ],
     outputs=[
@@ -124,7 +125,8 @@ gui = gr.Interface(
         gr.Textbox(type='text', label="Predictions (JSON)")
     ],
     examples=[
-        ["https://puam-loris.aws.princeton.edu/loris/INV33883.jp2/full/full/0/default.jpg", "", 0.05]
+        ["https://puam-loris.aws.princeton.edu/loris/INV33883.jp2/full/full/0/default.jpg", "", 0.05],
+        ["", "https://upload.wikimedia.org/wikipedia/commons/6/62/Gaspar_Peeter_Verbruggen_d.%C3%86._-_Blomsterkrans_med_Johannes_D%C3%B8beren_-_KMSsp332_-_Statens_Museum_for_Kunst.jpg", 0.35]
     ],
 )