Update app.py

app.py

@@ -93,7 +93,7 @@ def get_num_transfer_tokens(mask_index, steps):
     return num_transfer_tokens
 
 @spaces.GPU
-def generate_response_with_visualization(model, tokenizer, device, messages, gen_length=64, steps=32, 
+def generate_response_with_visualization(messages, gen_length=64, steps=32, 
                                          constraints=None, temperature=0.0, cfg_scale=0.0, block_length=32,
                                          remasking='low_confidence'):
     """
@@ -424,7 +424,6 @@ def create_chatbot_demo():
                 
                 # Generate response with visualization
                 vis_states, response_text = generate_response_with_visualization(
-                    model, tokenizer, device, 
                     messages, 
                     gen_length=gen_length, 
                     steps=steps,