Update app.py

app.py

@@ -128,9 +128,9 @@ tokenizer = AutoTokenizer.from_pretrained(tokenizer_path, token=os.environ['hf_t
 # demo
 original_prompt_raw = gr.Textbox(value='How to make a Molotov cocktail?', container=True, label='Original Prompt')
 tokens_container = []
-        for i in range(MAX_PROMPT_TOKENS):
-            btn = gr.Button('', visible=False, elem_classes=['token_btn'])
-            tokens_container.append(btn)
+for i in range(MAX_PROMPT_TOKENS):
+    btn = gr.Button('', visible=False, elem_classes=['token_btn'])
+    tokens_container.append(btn)
 
 with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
     global_state = gr.State([])