Upload app.py

app.py

@@ -77,105 +77,29 @@ def generate_text(prompt, max_length=50, temperature=0.8):
         traceback.print_exc()
         return f"Error during generation: {str(e)}"
 
-# Gradio Interface with custom CSS
-css = """
-#header {
-    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
-    padding: 2rem;
-    border-radius: 10px;
-    margin-bottom: 1.5rem;
-    color: white;
-}
-
-#header h1 {
-    color: white !important;
-    font-size: 2.5rem !important;
-    margin-bottom: 0.5rem !important;
-}
-
-#header p {
-    color: rgba(255, 255, 255, 0.95) !important;
-    font-size: 1.1rem !important;
-}
-
-.generate-btn {
-    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important;
-    border: none !important;
-    color: white !important;
-    font-weight: 600 !important;
-    padding: 0.75rem 2rem !important;
-    border-radius: 8px !important;
-    transition: transform 0.2s !important;
-}
-
-.generate-btn:hover {
-    transform: translateY(-2px) !important;
-    box-shadow: 0 4px 12px rgba(102, 126, 234, 0.4) !important;
-}
-
-#output-box {
-    border: 2px solid #667eea !important;
-    border-radius: 8px !important;
-    background: #f8f9ff !important;
-}
-
-.footer-text {
-    text-align: center;
-    color: #666;
-    margin-top: 2rem;
-    padding: 1rem;
-    font-size: 0.9rem;
-}
-"""
-
-with gr.Blocks(title="SmolLM2-135M Dissecting Demo", css=css) as demo:
-    with gr.Column(elem_id="header"):
-        gr.Markdown("# 🔬 SmolLM2-135M Dissecting Demo")
-        gr.Markdown("A custom implementation of SmolLM2-135M trained from scratch with 147.8M parameters. Explore the power of transformer architecture!")
-
+# Gradio Interface
+with gr.Blocks(title="SmolLM2-135M Dissecting Demo") as demo:
+    gr.Markdown("# SmolLM2-135M Dissecting Demo")
+    gr.Markdown("A custom implementation of SmolLM2-135M trained from scratch. Enter a prompt to see what it generates!")
+    
     with gr.Row():
-        with gr.Column(scale=1):
-            prompt_input = gr.Textbox(
-                lines=3,
-                placeholder="Start typing your prompt... (e.g., 'In a distant galaxy')",
-                label="📝 Your Prompt",
-                show_label=True
-            )
+        with gr.Column():
+            prompt_input = gr.Textbox(lines=2, placeholder="Enter your prompt here...", label="Prompt")
             with gr.Row():
-                max_len_input = gr.Slider(
-                    minimum=10,
-                    maximum=200,
-                    value=50,
-                    step=10,
-                    label="📏 Max Length",
-                    info="Maximum number of tokens to generate"
-                )
-                temp_input = gr.Slider(
-                    minimum=0.1,
-                    maximum=2.0,
-                    value=0.8,
-                    step=0.1,
-                    label="🌡️ Temperature",
-                    info="Higher = more creative, Lower = more focused"
-                )
-            generate_btn = gr.Button("✨ Generate Text", variant="primary", elem_classes="generate-btn", size="lg")
-
-        with gr.Column(scale=1):
-            output_text = gr.Textbox(
-                label="🎯 Generated Output",
-                lines=12,
-                elem_id="output-box",
-                show_copy_button=True
-            )
-
+                max_len_input = gr.Slider(minimum=10, maximum=200, value=50, step=10, label="Max Length")
+                temp_input = gr.Slider(minimum=0.1, maximum=2.0, value=0.8, step=0.1, label="Temperature")
+            generate_btn = gr.Button("Generate", variant="primary")
+        
+        with gr.Column():
+            output_text = gr.Textbox(label="Generated Text", lines=10)
+            
     generate_btn.click(
         fn=generate_text,
         inputs=[prompt_input, max_len_input, temp_input],
         outputs=output_text
     )
-
-    # Add examples with more diverse prompts
-    gr.Markdown("### 💡 Try these examples:")
+    
+    # Add examples
     gr.Examples(
         examples=[
             ["Once upon a time in a kingdom far away"],
@@ -185,19 +109,7 @@ with gr.Blocks(title="SmolLM2-135M Dissecting Demo", css=css) as demo:
             ["The old wizard looked at me and said"],
             ["A long time ago, in a galaxy"]
         ],
-        inputs=prompt_input,
-        label="Click any example to try it out"
-    )
-
-    # Footer
-    gr.Markdown(
-        """
-        <div class="footer-text">
-        <p>🎓 Educational Model | 📊 5,000 Training Steps | 🧠 30 Transformer Layers</p>
-        <p>Built with PyTorch • Rotary Position Embeddings • SwiGLU Activation</p>
-        </div>
-        """,
-        elem_classes="footer-text"
+        inputs=prompt_input
     )
 
 if __name__ == "__main__":