Update ui/app_class.py

ui/app_class.py

@@ -28,20 +28,24 @@ DESCRIPTION = """
         MINIMA: Modality Invariant Image Matching
     </h1>
 </div> 
+
 <div style="display: flex; justify-content: center; align-items: flex-start; flex-wrap: wrap; margin-bottom: 20px;">
     <div>
         <a href="https://github.com/LSXI7/MINIMA"><img src="https://img.shields.io/badge/Source_Code-black?logo=Github" alt="Github Source Code"></a>
     </div>&nbsp;
     <div>
-        <a href="https://github.com/LSXI7/MINIMA/blob/main/assets/MINIMA-arxiv-submitted.pdf"><img src="https://img.shields.io/badge/arXiv-PDF-b31b1b" alt="arXiv"></a>
+        <a href="https://arxiv.org/abs/0.0"><img src="https://img.shields.io/badge/arXiv-PDF-b31b1b" alt="arXiv"></a>
     </div>
 </div>
+
 <p style="text-align: center; font-size: 16px; color: #666;">
     This Space is derived from <a href="https://github.com/Vincentqyw/image-matching-webui" style="color: #007BFF;">Image Matching WebUI</a>.      
 </p>
+
 <p style="text-align: center; font-size: 16px; color: #666;">
     We are grateful to the authors for their contribution of the source code. 
 </p>
+
 <p style="text-align: center; font-size: 16px; color: #666;">
     Here we provide our MINIMA-model in our paper for test and comparison, and this project is undergoing continuous enhancement.
 </p>
@@ -56,9 +60,9 @@ body {
     padding: 0; 
 }
 #logo-img {
-    width: 220px; 
+    width: 100px; 
     height: auto; 
-    margin: 0;
+    margin: 20px auto;
     display: block;
 }
 h1 {
@@ -80,7 +84,6 @@ a:hover {
 """
 
 
-
 class ImageMatchingApp:
     def __init__(self, server_name="0.0.0.0", server_port=7860, **kwargs):
         self.server_name = server_name
@@ -105,7 +108,7 @@ class ImageMatchingApp:
         with gr.Blocks(css=CSS) as self.app:
             with gr.Tab("Image Matching"):
                 with gr.Row():
-                    with gr.Column(scale=1, min_width=150):
+                    with gr.Column(scale=1):
                         gr.Image(
                             str(
                                 Path(__file__).parent.parent
@@ -116,14 +119,14 @@ class ImageMatchingApp:
                             show_share_button=False,
                             show_download_button=False,
                         )
-                    with gr.Column(scale=6, min_width=300):
+                    with gr.Column(scale=3):
                         gr.HTML(DESCRIPTION)
                 with gr.Row(equal_height=False):
                     with gr.Column():
                         with gr.Row():
                             matcher_list = gr.Dropdown(
                                 choices=self.init_matcher_dropdown(),
-                                value="superpoint+minima(lightglue)",
+                                value="superpoint-minima_lightglue",
                                 label="Matching Model",
                                 interactive=True,
                             )
@@ -196,7 +199,7 @@ class ImageMatchingApp:
                                         maximum=10000,
                                         step=10,
                                         label="Max features",
-                                        value=1000,
+                                        value=2048,
                                     )
                                 # TODO: add line settings
                                 with gr.Row():
@@ -490,8 +493,10 @@ class ImageMatchingApp:
     def ui_change_imagebox(self, choice):
         """
         Updates the image box with the given choice.
+
         Args:
             choice (list): The list of image sources to be displayed in the image box.
+
         Returns:
             dict: A dictionary containing the updated value, sources, and type for the image box.
         """
@@ -541,6 +546,7 @@ class ImageMatchingApp:
     ]:
         """
         Reset the state of the UI.
+
         Returns:
             tuple: A tuple containing the initial values for the UI state.
         """
@@ -766,7 +772,7 @@ class AppSfmUI(AppBaseUI):
                                         label="Match Threshold",
                                         minimum=0.01,
                                         maximum=12.0,
-                                        value=0.,
+                                        value=0.1,
                                     )
                                     self.inputs.ransac_threshold = gr.Slider(
                                         label="Ransac Threshold",
@@ -782,7 +788,7 @@ class AppSfmUI(AppBaseUI):
                                         label="Ransac Confidence",
                                         minimum=0.01,
                                         maximum=1.0,
-                                        value=0.9999,
+                                        value=0.999,
                                         step=0.0001,
                                         interactive=True,
                                     )