Update app.py

app.py

@@ -22,7 +22,6 @@ st.set_page_config(
     layout="wide"
 )
 
-
 # ---------------------------------------------------------------------
 # 2) CUSTOM CSS / UI DESIGN
 # ---------------------------------------------------------------------
@@ -92,7 +91,8 @@ def load_llama_pipeline(model_id: str, device: str, token: str):
         model_id,
         use_auth_token=token,
         torch_dtype=torch.float16 if device == "auto" else torch.float32,
-        device_map=device
+        device_map=device,
+        low_cpu_mem_usage=True
     )
     text_gen_pipeline = pipeline(
         "text-generation",
@@ -174,7 +174,7 @@ if not hf_token:
     st.error("No HF_TOKEN found. Please set it in your environment.")
     st.stop()
 
-if st.button("\u270d Generate Promo Script"):
+if st.button("✍ Generate Promo Script"):
     if not prompt.strip():
         st.error("Please provide a concept first.")
     else:
@@ -195,7 +195,7 @@ st.markdown("---")
 st.subheader("🎵 Step 2: Generate Audio")
 audio_length = st.slider("Track Length (tokens)", 128, 1024, 512, 64)
 
-if st.button("\ud83c\udfa7 Create Audio"):
+if st.button("🎧 Create Audio"):
     if "final_script" not in st.session_state:
         st.error("Please generate a script first.")
     else: