Update app.py

app.py

@@ -6,7 +6,6 @@ model_name = "distilgpt2"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
-# Sample starting scenario and choices
 scenario = "You wake up in a strange forest. Paths lead in all directions."
 choices = [
     "Walk north into the dense forest.",
@@ -30,14 +29,12 @@ def update_scenario(choice_index):
     new_scenario = generate_continuation(chosen_action)
     return new_scenario
 
-# Gradio app
 with gr.Blocks() as app:
     with gr.Row():
         story_display = gr.Textbox(value=scenario, show_label=False, lines=7)
+        choice_buttons = [gr.Button(choice) for choice in choices]
     
-    with gr.Row():
-        for i, choice in enumerate(choices):
-            button = gr.Button(choice)
-            button.click(fn=update_scenario, data=i, outputs=story_display)
+    for i, button in enumerate(choice_buttons):
+        button.click(fn=lambda x=i: update_scenario(x), inputs=None, outputs=story_display)
 
 app.launch()