update app.py

app.py

@@ -38,7 +38,7 @@ def query(query_index, query_domain):
     dst_img_path = os.path.join(data_root, dst_data[0][query_index])
     img_paths = [dst_img_path]
     q_cl = dst_img_path.split('/')[-2]
-    captions = [f'Query: {q_cl}']
+    captions = [f'Query: {q_cl}'.title()]
     for s_domain, s_data in src_data_dict.items():
         _, top_n_matches_ids = s_data[1].kneighbors(dst_data[1][query_index:query_index+1])
         top_n_labels = s_data[0][2][top_n_matches_ids][0]
@@ -48,7 +48,7 @@ def query(query_index, query_domain):
         for p in src_img_pths:
             src_cl = p.split('/')[-2]
             src_file =  p.split('/')[-1]
-            captions.append(src_cl)
+            captions.append(src_cl.title())
     print(img_paths)
     return tuple([p for p in img_paths])+ tuple(captions)
 
@@ -60,9 +60,14 @@ with demo:
     slider = gr.Slider(0, 1000)
     image_button = gr.Button("Run")
 
-    gr.Markdown('# Query Image')
-    src_cap = gr.Label()
-    src_img = gr.Image()
+    with gr.Row():
+        gr.Markdown('# Query Image: \t\t\t\t                   ')
+        gr.Markdown('\t')
+        gr.Markdown('\t')
+        gr.Markdown('\t')
+        with gr.Column():
+            src_cap = gr.Label()
+            src_img = gr.Image()
 
 
     out_images = []