enforce model not null constraint

demo_watermark.py

@@ -706,7 +706,7 @@ def run_gradio(args, model=None, device=None, tokenizer=None):
         detect_btn.click(fn=detect_partial, inputs=[detection_input,session_args,session_tokenizer], outputs=[detection_result, session_args,session_tokenizer], api_name="detection")
 
         # State management logic
-        # update callbacks that change the state dict
+        # define update callbacks that change the state dict
         def update_model(session_state, value): session_state.model_name_or_path = value; return session_state
         def update_sampling_temp(session_state, value): session_state.sampling_temp = float(value); return session_state
         def update_generation_seed(session_state, value): session_state.generation_seed = int(value); return session_state
@@ -769,20 +769,35 @@ def run_gradio(args, model=None, device=None, tokenizer=None):
             #     return ALPACA_MODEL_TOKENIZER.from_pretrained(ALPACA_TOKENIZER_PATH)
             # else:
             return AutoTokenizer.from_pretrained(model_name_or_path)
+        
+        def check_model(value): return value if (value!="" and value is not None) else args.model_name_or_path
+        # enforce constraint that model cannot be null or empty
+        # then attach model callbacks in particular
+        model_selector.change(check_model, inputs=[model_selector], outputs=[model_selector]).then(
+            toggle_vis_for_api_model,inputs=[model_selector], outputs=[n_beams]
+        ).then(
+            toggle_beams_for_api_model,inputs=[model_selector,n_beams], outputs=[n_beams]
+        ).then(
+            toggle_interactive_for_api_model,inputs=[model_selector], outputs=[gamma]
+        ).then(
+            toggle_interactive_for_api_model,inputs=[model_selector], outputs=[delta]
+        ).then(
+            toggle_gamma_for_api_model,inputs=[model_selector,gamma], outputs=[gamma]
+        ).then(
+            toggle_delta_for_api_model,inputs=[model_selector,delta], outputs=[delta]
+        ).then(
+            update_tokenizer,inputs=[model_selector], outputs=[session_tokenizer]
+        ).then(
+            update_model,inputs=[session_args, model_selector], outputs=[session_args]
+        ).then(
+            lambda value: str(value), inputs=[session_args], outputs=[current_parameters]
+        )
         # registering callbacks for toggling the visibilty of certain parameters based on the values of others
         decoding.change(toggle_sampling_vis,inputs=[decoding], outputs=[sampling_temp])
         decoding.change(toggle_sampling_vis,inputs=[decoding], outputs=[generation_seed])
         decoding.change(toggle_sampling_vis_inv,inputs=[decoding], outputs=[n_beams])
-        model_selector.change(toggle_vis_for_api_model,inputs=[model_selector], outputs=[n_beams])
         decoding.change(toggle_vis_for_api_model,inputs=[model_selector], outputs=[n_beams])
-        model_selector.change(toggle_beams_for_api_model,inputs=[model_selector,n_beams], outputs=[n_beams])
-        model_selector.change(toggle_interactive_for_api_model,inputs=[model_selector], outputs=[gamma])
-        model_selector.change(toggle_interactive_for_api_model,inputs=[model_selector], outputs=[delta])
-        model_selector.change(toggle_gamma_for_api_model,inputs=[model_selector,gamma], outputs=[gamma])
-        model_selector.change(toggle_delta_for_api_model,inputs=[model_selector,delta], outputs=[delta])
-        model_selector.change(update_tokenizer,inputs=[model_selector], outputs=[session_tokenizer])
         # registering all state update callbacks
-        model_selector.change(update_model,inputs=[session_args, model_selector], outputs=[session_args])
         decoding.change(update_decoding,inputs=[session_args, decoding], outputs=[session_args])
         sampling_temp.change(update_sampling_temp,inputs=[session_args, sampling_temp], outputs=[session_args])
         generation_seed.change(update_generation_seed,inputs=[session_args, generation_seed], outputs=[session_args])
@@ -798,7 +813,6 @@ def run_gradio(args, model=None, device=None, tokenizer=None):
         # register additional callback on button clicks that updates the shown parameters window
         generate_btn.click(lambda value: str(value), inputs=[session_args], outputs=[current_parameters])
         detect_btn.click(lambda value: str(value), inputs=[session_args], outputs=[current_parameters])
-        model_selector.change(lambda value: str(value), inputs=[session_args], outputs=[current_parameters])
         # When the parameters change, display the update and also fire detection, since some detection params dont change the model output.
         delta.change(lambda value: str(value), inputs=[session_args], outputs=[current_parameters])
         gamma.change(lambda value: str(value), inputs=[session_args], outputs=[current_parameters])