Rename labels

app.py

@@ -224,9 +224,9 @@ Some example images are provided at the end of this page.
             # inp_slider = gr.Slider(0, 100, value=50, label="Mask ratio", info="Choose ratio of masking between 0 and 100", elem_id='slider'),
             btn = gr.Button("Submit")
     with gr.Row():
-        gr.Markdown(value='## Original images')
+        gr.Markdown(value='## Input time series')
         gr.Markdown(value='## Masked images')
-        gr.Markdown(value='## Reconstructed images (with unmasked patches)')
+        gr.Markdown(value='## Reconstructed images*')
 
     original = []
     masked = []
@@ -242,6 +242,8 @@ Some example images are provided at the end of this page.
                 masked.append(gr.Image(image_mode='RGB', show_label=False, show_fullscreen_button=False))
                 predicted.append(gr.Image(image_mode='RGB', show_label=False, show_fullscreen_button=False))
 
+    gr.Markdown(value='* The reconstructed images include the ground truth unmasked patches.')
+
     btn.click(fn=run_inference,
               inputs=inp_files,
               outputs=original + masked + predicted)