updated

app.py

@@ -16,7 +16,7 @@ MODEL_CONFIG = {
     'cost_per_hour': 1.8,  # $1.8 per hour
 }
 
-original_time_cache = {22.57}
+original_time_cache = {"original_time": 22.57}
 
 
 def set_seed(seed: int = 42):
@@ -115,8 +115,7 @@ def load_original_model():
         model = HFMusicgenForConditionalGeneration.from_pretrained(
             "facebook/musicgen-large",
             torch_dtype=torch.float16,
-            device=device,
-        )
+        ).to(device)
 
         model.eval()
 
@@ -263,7 +262,6 @@ def get_cache_key(prompt, duration, guidance_scale):
 
 def generate_music_batch(text_prompt, duration=10, guidance_scale=3.0, model_mode="compressed"):
     try:
-        cache_key = get_cache_key(text_prompt, duration, guidance_scale)
 
         generator, processor = load_model()
         model_name = "Compressed (S)"
@@ -333,8 +331,8 @@ def generate_music_batch(text_prompt, duration=10, guidance_scale=3.0, model_mod
 
         comparison_message = ""
 
-        if cache_key in original_time_cache:
-            original_time = original_time_cache[cache_key]
+        if "original_time" in original_time_cache:
+            original_time = original_time_cache["original_time"]
             cost_info = calculate_cost_savings(generation_time, original_time)
 
             comparison_message = f"💰 Cost Savings: ${cost_info['savings']:.4f} ({cost_info['savings_percent']:.1f}%) - Compressed: ${cost_info['compressed_cost']:.4f} vs Original: ${cost_info['original_cost']:.4f}"