updated

app.py

@@ -16,7 +16,7 @@ MODEL_CONFIG = {
     'cost_per_hour': 1.8,  # $1.8 per hour
 }
 
-original_time_cache = {}
+original_time_cache = {22.57}
 
 
 def set_seed(seed: int = 42):
@@ -115,7 +115,8 @@ def load_original_model():
         model = HFMusicgenForConditionalGeneration.from_pretrained(
             "facebook/musicgen-large",
             torch_dtype=torch.float16,
-        ).to(device)
+            device=device,
+        )
 
         model.eval()