Update app.py

app.py

@@ -40,18 +40,32 @@ def generate(prompt, temperature=0.9, max_new_tokens=256, top_p=0.95, repetition
         yield output.strip('</s>')
     return output.strip('</s>')
 
-with gr.Blocks(css="""
-    body {background-color: #121212; color: #ffffff;}
-    .gradio-container {background-color: #121212; color: #ffffff;}
-    input, textarea, select, button {background-color: #333333 !important; color: #ffffff !important; border-color: #555555 !important;}
-    .gradio-container button {background-color: #555555 !important;}
-    """) as demo:
-    
-    with gr.Row():
-        input_text = gr.Textbox(placeholder="Enter your prompt here...", lines=2, max_lines=2, label="Prompt")
-        submit_button = gr.Button("Generate")
-        output_text = gr.Textbox(label="Output", interactive=True, lines=10)
-    
-    submit_button.click(fn=generate, inputs=input_text, outputs=output_text)
-
-demo.launch()
+def dark_theme():
+    return """
+        body {
+            background-color: #121212;
+            color: #ffffff;
+        }
+        .gradio-container {
+            background-color: #121212;
+            color: #ffffff;
+        }
+        input, textarea, select {
+            background-color: #333333 !important;
+            color: #ffffff !important;
+            border-color: #555555 !important;
+        }
+        .gradio-container button {
+            background-color: #555555 !important;
+            color: #ffffff !important;
+        }
+    """
+
+with gr.Interface(
+    fn=generate,
+    inputs=gr.Textbox(placeholder="Enter your prompt here...", label="Prompt"),
+    outputs=gr.Textbox(label="Output", interactive=True, lines=10),
+    theme="compact",  # Use compact theme for default layout
+    css=dark_theme()  # Apply dark theme styles
+) as app:
+    app.launch()