Update app.py

app.py

@@ -25,6 +25,7 @@ if file_name is not None:
     
     image = Image.open(file_name)
     input_image, copy_img= process_img(image)
+    col1.header("Input Image")
     col1.image(input_image, use_column_width=True)
 
     bbox,confidance,classes,nc = model(input_image)
@@ -45,7 +46,7 @@ if file_name is not None:
             if class_name =="Table":
                 color = (255,0,0) #Red color
                 cv2.rectangle(copy_img, (int(x1), int(y1)), (int(x2), int(y2)),color, 2)
-
+    col2.header("output Result")
     col2.image(copy_img, use_column_width=True)