Delete generate.py

generate.py

@@ -1,39 +0,0 @@
-import torch
-from transformers import AutoProcessor, MusicgenForConditionalGeneration
-import soundfile as sf
-import uuid
-
-MODEL_NAME = "facebook/musicgen-small"
-
-print("Loading MusicGen model...")
-
-processor = AutoProcessor.from_pretrained(MODEL_NAME)
-model = MusicgenForConditionalGeneration.from_pretrained(MODEL_NAME)
-
-device = "cuda" if torch.cuda.is_available() else "cpu"
-model.to(device)
-
-def generate_music(prompt, duration):
-
-    inputs = processor(
-        text=[prompt],
-        padding=True,
-        return_tensors="pt"
-    ).to(device)
-
-    audio_values = model.generate(
-        **inputs,
-        max_new_tokens=int(duration * 50)
-    )
-
-    filename = f"/tmp/{uuid.uuid4()}.wav"
-
-    sampling_rate = model.config.audio_encoder.sampling_rate
-
-    sf.write(
-        filename,
-        audio_values[0, 0].cpu().numpy(),
-        sampling_rate
-    )
-
-    return filename