initial

app.py

@@ -197,7 +197,7 @@ def generate_viz_wrapper_t2i(prompt_text, steps, guidance_scale, mask_schedule="
 
 @torch.no_grad()
 def generate_viz_wrapper_lm(prompt_text, steps, gen_length, block_length, temperature,
-                         cfg_scale, remasking_strategy, thinking_mode_lm): 
+                         cfg_scale, remasking_strategy, thinking_mode_lm=False): 
     global MODEL, TOKENIZER, MASK_ID, DEVICE
     print(f"thinking_mode_lm: {thinking_mode_lm}")
     if MODEL is None or TOKENIZER is None or MASK_ID is None:
@@ -350,7 +350,7 @@ def generate_viz_wrapper_lm(prompt_text, steps, gen_length, block_length, temper
 
 @torch.no_grad()
 def generate_viz_wrapper(uploaded_image_pil, prompt_text, steps, gen_length, block_length, temperature,
-                         cfg_scale, remasking_strategy, thinking_mode_mmu): 
+                         cfg_scale, remasking_strategy, thinking_mode_mmu=False): 
     global MODEL, TOKENIZER, MASK_ID, DEVICE
 
     if MODEL is None or TOKENIZER is None or MASK_ID is None:
@@ -868,4 +868,4 @@ with gr.Blocks(css=css_styles, theme=theme) as demo:
 
 if __name__ == "__main__":
     print(f"Starting Gradio App. Attempting to use device: {DEVICE}")
-    demo.launch(share=True)
+    demo.launch()