Update app.py

app.py

@@ -118,43 +118,59 @@ def main():
                        "2025"]
 
     with gr.Blocks(title=TITLE) as demo:
-        gr.Markdown(f"<h1 style='text-align: center;'>{TITLE}</h1>")
-        gr.Markdown(DESCRIPTION)
+    gr.Markdown(f"<h1 style='text-align: center;'>{TITLE}</h1>")
+    gr.Markdown(DESCRIPTION)
+
+    with gr.Row():
+        with gr.Column():
+
+            submit = gr.Button(
+                value="Process Images", variant="primary"
+            )
+
+            image_files = gr.File(
+                file_types=["image"], label="Upload Images", file_count="multiple",
+            )
+
+            # Wrap the model selection and sliders in an Accordion
+            with gr.Accordion("Advanced Settings", open=False):  # Collapsible by default
+                model_repo = gr.Dropdown(
+                    model_repos,
+                    value=VIT_MODEL_DSV3_REPO,
+                    label="Select Model",
+                )
+
+                # Ensure args.score_slider_step and args.score_general_threshold are valid floats
+                if isinstance(args.score_slider_step, bool):
+                    args.score_slider_step = 0.1  # Default value if it's a boolean
+                if isinstance(args.score_general_threshold, bool):
+                    args.score_general_threshold = 0.25  # Default value if it's a boolean
+
+                general_thresh = gr.Slider(
+                    0, 1, step=float(args.score_slider_step or 0.1), 
+                    value=float(args.score_general_threshold or 0.25), 
+                    label="General Tags Threshold"
+                )
 
-        with gr.Row():
-            with gr.Column():
+                # Ensure args.score_character_threshold is a valid float
+                if isinstance(args.score_character_threshold, bool):
+                    args.score_character_threshold = 1.0  # Default value if it's a boolean
 
-                submit = gr.Button(
-                    value="Process Images", variant="primary"
+                character_thresh = gr.Slider(
+                    0, 1, step=float(args.score_slider_step or 0.1), 
+                    value=float(args.score_character_threshold or 1.0), 
+                    label="Character Tags Threshold"
                 )
-                
-                image_files = gr.File(
-                    file_types=["image"], label="Upload Images", file_count="multiple",
+
+                filter_tags = gr.Textbox(
+                    value=", ".join(predefined_tags),
+                    label="Filter Tags (comma-separated)",
+                    placeholder="Add tags to filter out (e.g., winter, red, from above)",
+                    lines=9
                 )
-                
-                # Wrap the model selection and sliders in an Accordion
-                with gr.Accordion("Advanced Settings", open=False):  # Collapsible by default
-                    model_repo = gr.Dropdown(
-                        model_repos,
-                        value=VIT_MODEL_DSV3_REPO,
-                        label="Select Model",
-                    )
-                    general_thresh = gr.Slider(
-                        0, 1, step=args.score_slider_step, value=args.score_general_threshold, label="General Tags Threshold"
-                    )
-                    character_thresh = gr.Slider(
-                        0, 1, step=args.score_slider_step, value=args.score_character_threshold, label="Character Tags Threshold"
-                    )
-                    filter_tags = gr.Textbox(
-                        value=", ".join(predefined_tags),
-                        label="Filter Tags (comma-separated)",
-                        placeholder="Add tags to filter out (e.g., winter, red, from above)",
-                        lines=9
-                    )
-
-
-            with gr.Column():
-                output = gr.Textbox(label="Output", lines=10)
+
+        with gr.Column():
+            output = gr.Textbox(label="Output", lines=10)
 
         def process_images(files, model_repo, general_thresh, character_thresh, filter_tags):
             images = [Image.open(file.name) for file in files]