Update app.py

app.py

@@ -14,9 +14,9 @@ pipe = pipeline(
 
 # HTML content
 HTML_CONTENT = """
-<h1 style="text-align: center;">Llama-3.1-Storm-8B Text Generation</h1>
-<p style="text-align: center;">Generate text using the powerful Llama-3.1-Storm-8B model. Enter a prompt or select an example, and let the AI create!</p>
-<div style="display: flex; justify-content: center; margin-bottom: 20px;">
+<h1>Llama-3.1-Storm-8B Text Generation</h1>
+<p>Generate text using the powerful Llama-3.1-Storm-8B model. Enter a prompt and let the AI create!</p>
+<div class="llama-image">
     <img src="https://cdn-uploads.huggingface.co/production/uploads/64c75c1237333ccfef30a602/tmOlbERGKP7JSODa6T06J.jpeg" alt="Llama" style="width:200px; border-radius:10px;">
 </div>
 """
@@ -39,11 +39,11 @@ def generate_text(prompt, max_length, temperature):
     
     return outputs[0]['generated_text'][len(formatted_prompt):]
 
+# Define examples
 examples = [
-    "Write a short story about a magical llama.",
-    "Explain the concept of machine learning to a 10-year-old.",
-    "Describe the process of making the perfect cup of coffee.",
-    "What are the main differences between Python and JavaScript?"
+    ["Tell me a short story about a brave astronaut.", 150, 0.7],
+    ["Explain quantum computing in simple terms.", 200, 0.5],
+    ["Write a haiku about spring.", 50, 1.0],
 ]
 
 with gr.Blocks() as demo:
@@ -57,13 +57,13 @@ with gr.Blocks() as demo:
         with gr.Column(scale=2):
             output = gr.Textbox(label="Generated Text", lines=10)
     
+    submit_button.click(generate_text, inputs=[prompt, max_length, temperature], outputs=[output])
+    
+    # Add examples
     gr.Examples(
         examples=examples,
-        inputs=prompt,
-        label="Click on an example to load it into the prompt box:"
+        inputs=[prompt, max_length, temperature],
     )
-    
-    submit_button.click(generate_text, inputs=[prompt, max_length, temperature], outputs=[output])
 
 if __name__ == "__main__":
     demo.launch()