tried to fix model loading bug

app.py

@@ -10,6 +10,7 @@ from PIL import Image
 import spaces
 
 
+
 def image_transform(image, resolution=256, normalize=True):
     image = transforms.Resize(resolution, interpolation=transforms.InterpolationMode.BICUBIC)(image)
     image = transforms.CenterCrop((resolution, resolution))(image)
@@ -259,7 +260,7 @@ def get_highlighted_text_tuples(current_x_ids_batch, prompt_input_ids, prompt_le
 @torch.no_grad()
 @spaces.GPU
 def generate_viz_wrapper_t2i(prompt_text, steps, guidance_scale, mask_schedule="cosine"):
-    global MODEL, TOKENIZER, MASK_ID, DEVICE, uni_prompting
+    global MODEL, TOKENIZER, MASK_ID, DEVICE, uni_prompting, VQ_MODEL
 
     if MODEL is None or TOKENIZER is None or MASK_ID is None:
         yield [("Error: Model not loaded. Please load the model first.", "ERROR")], "Model not loaded."
@@ -315,7 +316,7 @@ def generate_viz_wrapper_t2i(prompt_text, steps, guidance_scale, mask_schedule="
 @spaces.GPU
 def generate_viz_wrapper_lm(prompt_text, steps, gen_length, block_length, temperature,
                          cfg_scale, remasking_strategy, thinking_mode_lm=False): 
-    global MODEL, TOKENIZER, MASK_ID, DEVICE
+    global MODEL, TOKENIZER, MASK_ID, DEVICE, VQ_MODEL
     if MODEL is None or TOKENIZER is None or MASK_ID is None:
         yield [("Error: Model not loaded. Please load the model first.", "ERROR")], "Model not loaded."
         return
@@ -1130,9 +1131,9 @@ with gr.Blocks(css=css_styles, theme=theme) as demo:
         return None, None, None  # Clear image, visualization, and final text
 
     clear_button_ui_lm.click(
-        fn=clear_outputs,
+        fn=lambda: (None, None), # 返回两个 None
         inputs=None,
-        outputs=[image_upload_box, output_visualization_box_lm, output_final_text_box_lm],
+        outputs=[output_visualization_box_lm, output_final_text_box_lm], # 只清除两个文本框
         queue=False
     )
     clear_button_ui_mmu.click(