Update app.py

app.py

@@ -1,4 +1,4 @@
-import os
+# import os
 import re
 import torch
 import tempfile
@@ -205,9 +205,7 @@ def generate_music(prompt: str, audio_length: int):
         musicgen_model, musicgen_processor = get_musicgen_model(model_key)
 
         device = "cuda" if torch.cuda.is_available() else "cpu"
-        # Process the input and move each tensor to the proper device
-        inputs = musicgen_processor(text=[prompt], padding=True, return_tensors="pt")
-        inputs = {k: v.to(device) for k, v in inputs.items()}
+        inputs = musicgen_processor(text=[prompt], padding=True, return_tensors="pt").to(device)
 
         with torch.inference_mode():
             outputs = musicgen_model.generate(**inputs, max_new_tokens=audio_length)
@@ -316,12 +314,12 @@ with gr.Blocks(css="""
     # Custom Header
     with gr.Row(elem_classes="header"):
         gr.Markdown("""
-        <h1>🎧 AI Promo Studio</h1>
-        <p>Your all-in-one AI solution for crafting engaging audio promos.</p>
+        <h1>🎧 AI Ads Promo</h1>
+        <p>Your all-in-one AI solution for crafting engaging audio ads. <br><em>Demo MVP</em></p>
         """)
 
     gr.Markdown("""
-    Welcome to **AI Promo Studio**! This platform leverages state-of-the-art AI models to help you generate:
+    Welcome to **AI Ads Promo (Demo MVP)**! This platform leverages state-of-the-art AI models to help you generate:
     
     - **Script**: Generate a compelling voice-over script with LLaMA.
     - **Voice Synthesis**: Create natural-sounding voice-overs using Coqui TTS.
@@ -430,7 +428,7 @@ with gr.Blocks(css="""
         <hr>
         Created with ❤️ by <a href="https://bilsimaging.com" target="_blank" style="color: #88aaff;">bilsimaging.com</a>
         <br>
-        <small>AI Promo Studio &copy; 2025</small>
+        <small>AI Ads Promo (Demo MVP) &copy; 2025</small>
     </div>
     """)