Update app.py

app.py

@@ -29,15 +29,13 @@ def pdf_to_img(pdf_path):
 def concate_image(img_input):
     img_name_lst = img_input.split(',')
     if len(img_name_lst) == 1 :
-        img1 = cv2.imread(img_name_lst[0]) 
-        img2 = cv2.imread(img_name_lst[0])
-        im_v = cv2.vconcat([img1, img2]) 
+        return img_name_lst[0]
     else:
         img1 = cv2.imread(img_name_lst[0]) 
         img2 = cv2.imread(img_name_lst[1])
         im_v = cv2.vconcat([img1, img2])
-    cv2.imwrite('merge_image.jpg', im_v)
-    return "merge_image.jpg"
+        cv2.imwrite('merge_image.jpg', im_v)
+        return "merge_image.jpg"
 
 # def encode_image(image_path):
 #     file_path = concate_image(image_path)
@@ -107,8 +105,8 @@ with block:
         with gr.Row():
             with gr.Column():
                 image_name = gr.Textbox(label="""
-                Provide the Image names separated by a comma(,) \n
-                Note: Provide the name of the images to be compared.
+                Note: Provide the name of the images to be compared separated by a comma(,),
+                 
                 """)
                 show_button = gr.Button(value="Show Images")
                 output_img = gr.Image(type="filepath")