Update app.py

app.py

@@ -77,12 +77,17 @@ print("Model loaded successfully!")
 # Define the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("""
-        # CPU-Optimized StableLM Athena-1-1.5B Chat
-        Running with CPU optimizations for memory efficiency.
+        # Athena-1.1.5B Chat
+        Small, Smart, Super.
     """)
     
-    # Initialize state
+    # Initialize state and components
     state = gr.State("")
+    user_input = gr.Textbox(
+        label="Your Message",
+        placeholder="Type your question here...",
+        lines=3
+    )
     
     with gr.Row():
         with gr.Column(scale=2):
@@ -99,19 +104,14 @@ with gr.Blocks() as demo:
                 "Escribe un poema corto sobre la historia del Mediterráneo.",
                 "Jane has 8 apples. Solve this in Python."
             ]
+            example_buttons = []
             for example in examples:
-                gr.Button(example).click(
+                example_button = gr.Button(example)
+                example_button.click(
                     lambda e=example: (e, state.value),
                     outputs=[user_input, state]
                 )
-    
-    with gr.Row():
-        user_input = gr.Textbox(
-            label="Your Message",
-            placeholder="Type your question here...",
-            lines=3
-        )
-        submit_button = gr.Button("Submit")
+                example_buttons.append(example_button)
     
     output_text = gr.Textbox(
         label="Athena's Response",
@@ -119,6 +119,8 @@ with gr.Blocks() as demo:
         interactive=False
     )
     
+    submit_button = gr.Button("Submit")
+    
     # Handle submission
     submit_button.click(
         generate_response,