Update app.py

app.py

@@ -158,8 +158,8 @@ sample_images = get_sample_images('example_images')
 
 tab1 = gr.Interface(
     fn=predict_on_train,
-    inputs=gr.Image(label='', type="pil"),
-    outputs=[gr.Image(type="pil"), gr.Image(type="pil")],
+    inputs=gr.Image(),
+    outputs=[gr.Image(), gr.Image()],
     title='Images with Ground Truth',
     examples=sample_images
 )
@@ -167,8 +167,8 @@ tab1 = gr.Interface(
 # Create the video processing interface
 tab2 = gr.Interface(
     fn=predict_on_test,
-    inputs=gr.File(label=""),
-    outputs=gr.File(label=""),
+    inputs=gr.Image(),
+    outputs=gr.Image(),
     title='Images with Ground Truth',
     examples=sample_images
 )