Update src/demo/gradio_utils.py

src/demo/gradio_utils.py

@@ -17,7 +17,7 @@ if pkg_version != "0.25.1":
     import subprocess
     print(f"Installing huggingface_hub==0.25.1 (Current version: {pkg_version})")
     subprocess.check_call([sys.executable, "-m", "pip", "install", f"huggingface_hub==0.25.1"])
-    
+
 from src.demo.utils import (
     create_model, get_spec_pil, get_mask_region, get_mask_regions, func_clear,
     update_reference_spec, get_spec_pil_with_original, get_spec_pils_for_moving,
@@ -272,7 +272,7 @@ def create_remove_demo():
 
     # Claim variables
     config = gr.State(value=cfgs)
-    model = gr.State(value=None)
+    model = gr.State(value=MODEL)
     foreground_fbank = gr.State(value=None)
     background_fbank = gr.State(value=None)
     foreground_spec_plot_ori = gr.State(value=None)
@@ -444,7 +444,7 @@ def create_move_demo():
 
     # Claim variables
     config = gr.State(value=cfgs)
-    model = gr.State(value=None)
+    model = gr.State(value=MODEL)
     foreground_fbank = gr.State(value=None)
     background_fbank = gr.State(value=None)
     foreground_spec_plot_ori = gr.State(value=None)