Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from zeroshot import process, ZS_EXAMPLES
+from zeroshot import process, ZS_EXAMPLES, WORD_SCORE_DEFAULT_IF_LM, WORD_SCORE_DEFAULT_IF_NOLM, LM_SCORE_DEFAULT
 
 with gr.Blocks(css="style.css") as demo:
     gr.Markdown(
@@ -21,30 +21,54 @@ with gr.Blocks(css="style.css") as demo:
                     "The following parameters are used for beam-search decoding. Use the default values if you are not sure."
                 )
                 with gr.Row():
-                    wscore = gr.Slider(
-                        minimum=-10.0,
-                        maximum=10.0,
-                        value=0,
-                        step=0.1,
-                        interactive=True,
-                        label="Word Insertion Score",
-                    )
-                    lmscore = gr.Slider(
-                        minimum=-10.0,
-                        maximum=10.0,
-                        value=0,
-                        step=0.1,
-                        interactive=True,
-                        label="Language Model Score",
-                    )
-                with gr.Row():
-                    wscore_usedefault = gr.Checkbox(
-                        label="Use Default Word Insertion Score", value=True
-                    )
-                    lmscore_usedefault = gr.Checkbox(
-                        label="Use Default Language Model Score", value=True
-                    )
+                    with gr.Column():
+                        wscore_usedefault = gr.Checkbox(
+                            label="Use Default Word Insertion Score", value=True
+                        )
+                        wscore = gr.Slider(
+                            minimum=-10.0,
+                            maximum=10.0,
+                            value=WORD_SCORE_DEFAULT_IF_NOLM,
+                            step=0.1,
+                            interactive=False,
+                            label="Word Insertion Score",
+                        )
+                        
+                    with gr.Column():
+                        lmscore_usedefault = gr.Checkbox(
+                            label="Use Default Language Model Score", value=True
+                        )
+                        lmscore = gr.Slider(
+                            minimum=-10.0,
+                            maximum=10.0,
+                            value=0,
+                            step=0.1,
+                            interactive=False,
+                            label="Language Model Score",
+                        )
             btn = gr.Button("Submit", elem_id="submit")
+            
+            @gr.on(inputs=[wscore_usedefault, lmscore_usedefault, lm_file], outputs=[wscore, lmscore])
+            def update_slider(ws, ls, lm ):
+
+                ws_slider = gr.Slider(
+                    minimum=-10.0,
+                    maximum=10.0,
+                    value=LM_SCORE_DEFAULT if lm is not None else 0,
+                    step=0.1,
+                    interactive=not ws,
+                    label="Word Insertion Score",
+                )
+                ls_slider = gr.Slider(
+                    minimum=-10.0,
+                    maximum=10.0,
+                    value=WORD_SCORE_DEFAULT_IF_NOLM if lm is None else WORD_SCORE_DEFAULT_IF_LM,
+                    step=0.1,
+                    interactive=not ls,
+                    label="Language Model Score",
+                )
+                return ws_slider, ls_slider
+    
         with gr.Column():
             text = gr.Textbox(label="Transcript")
     btn.click(