Update app.py

app.py

@@ -165,10 +165,10 @@ with gr.Blocks(css= "style.css") as app:
                 # Components
                 image = gr.Image(type="pil", image_mode="RGBA", label="Input")
                 submit_button = gr.Button(value="Submit", variant="primary", size="lg")
-                model_repo = gr.Dropdown(dropdown_list, value=SWINV2_MODEL_DSV3_REPO, label="Model")
-                general_thresh = gr.Slider(0, 1, step=args.score_slider_step, value=args.score_general_threshold, label="General Tags Threshold", scale=3)
+                model_repo = gr.Dropdown(modules.wdtagger.dropdown_list, value=modules.wdtagger.dropdown_list[0], label="Model")
+                general_thresh = gr.Slider(0, 1, step=modules.wdtagger.args.score_slider_step, value=args.score_general_threshold, label="General Tags Threshold", scale=3)
                 general_mcut_enabled = gr.Checkbox(value=False, label="Use MCut threshold", scale=1)
-                character_thresh = gr.Slider(0, 1, step=args.score_slider_step, value=args.score_character_threshold, label="Character Tags Threshold", scale=3)
+                character_thresh = gr.Slider(0, 1, step=modules.wdtagger.args.score_slider_step, value=args.score_character_threshold, label="Character Tags Threshold", scale=3)
                 character_mcut_enabled = gr.Checkbox(value=False, label="Use MCut threshold", scale=1)
                 clear_button = gr.ClearButton(components=[image, model_repo, general_thresh, general_mcut_enabled, character_thresh, character_mcut_enabled], variant="secondary", size="lg")
                 
@@ -179,7 +179,7 @@ with gr.Blocks(css= "style.css") as app:
                 general_res = gr.Label(label="Output (tags)")
         
         clear_button.add([sorted_general_strings, rating, character_res, general_res])
-        submit_button.click(predictor.predict, inputs=[image, model_repo, general_thresh, general_mcut_enabled, character_thresh, character_mcut_enabled], outputs=[sorted_general_strings, rating, character_res, general_res])
+        submit_button.click(modules.wdtagger.predictor.predict, inputs=[image, model_repo, general_thresh, general_mcut_enabled, character_thresh, character_mcut_enabled], outputs=[sorted_general_strings, rating, character_res, general_res])
 
 # Jalankan antarmuka
 app.launch()