Update app.py

app.py

@@ -197,7 +197,7 @@ description= '''
 tab1 = gr.Interface(
     fn=predict_on_train,
     inputs=[gr.Image(), gr.Image()],
-    outputs=[gr.Image(), gr.Image()],
+    outputs=[gr.Image(label='Ground Truth'), gr.Image(label='Predicted')],
     title='Images with Ground Truth',
     description=description,
     examples=train_images
@@ -207,7 +207,7 @@ tab1 = gr.Interface(
 tab2 = gr.Interface(
     fn=predict_on_test,
     inputs=gr.Image(),
-    outputs=gr.Image(),
+    outputs=gr.Image(label='Predicted'),
     title='Images without Ground Truth',
     description=description,
     examples=test_images