Update app.py

app.py

@@ -39,6 +39,7 @@ def clean_text(text: str) -> str:
     """
     Removes undesired characters (e.g., asterisks) that might not be recognized by the model's vocabulary.
     """
+    # Remove all asterisks. You can add more cleaning steps here as needed.
     return re.sub(r'\*', '', text)
 
 # ---------------------------------------------------------------------
@@ -74,6 +75,7 @@ def get_musicgen_model(model_key: str = "facebook/musicgen-large"):
 
     model = MusicgenForConditionalGeneration.from_pretrained(model_key)
     processor = AutoProcessor.from_pretrained(model_key)
+
     device = "cuda" if torch.cuda.is_available() else "cpu"
     model.to(device)
     MUSICGEN_MODELS[model_key] = (model, processor)
@@ -203,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)