zerpgpu

app.py

@@ -1,3 +1,4 @@
+import spaces  # Wichtig: Modul importieren, um ZeroGPU zu aktivieren
 import gradio as gr
 from transformers import pipeline
 import scipy.io.wavfile
@@ -6,18 +7,23 @@ import os
 # Sicherstellen, dass der 'static'-Ordner existiert
 os.makedirs('static', exist_ok=True)
 
-# MusicGen-Pipeline initialisieren
-synthesizer = pipeline(
-    "text-to-audio",
-    "facebook/musicgen-small",
-    model_kwargs={"attn_implementation": "eager"}
-)
+# Funktion zur Initialisierung des Synthesizers auf der GPU
+@spaces.GPU
+def initialize_synthesizer():
+    return pipeline(
+        "text-to-audio",
+        "facebook/musicgen-small",
+        model_kwargs={"attn_implementation": "eager"}
+    )
+
+# Synthesizer initialisieren
+synthesizer = initialize_synthesizer()
 
-# Funktion zur Musikgenerierung
+# GPU-abhängige Funktion zur Musikgenerierung
+@spaces.GPU(duration=120)  # GPU wird für diese Funktion angefordert
 def generate_music(prompt, duration, diffusion_steps, cfg_scale):
     try:
-        # Placeholder für generierte Musik
-        # Da MusicGen keine Diffusion Steps und CFG Scale unterstützt, sind diese Parameter hier symbolisch.
+        # Musik generieren
         music = synthesizer(prompt, forward_params={"do_sample": True})
         
         # Datei speichern