up

handler.py

@@ -2,17 +2,6 @@ from typing import Dict, List, Any
 from transformers import AutoProcessor, MusicgenForConditionalGeneration
 import torch
 
-processor = AutoProcessor.from_pretrained("facebook/musicgen-small")
-model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-small")
-
-inputs = processor(
-    text=["80s pop track with bassy drums and synth", "90s rock song with loud guitars and heavy drums"],
-    padding=True,
-    return_tensors="pt",
-)
-
-audio_values = model.generate(**inputs, max_new_tokens=256)
-
 class EndpointHandler:
     def __init__(self, path=""):
         # load model and processor from path