zerov2

app.py

@@ -1,4 +1,4 @@
-import spaces  # Wichtig: Modul importieren, um ZeroGPU zu aktivieren
+import spaces  # Für ZeroGPU
 import gradio as gr
 from transformers import pipeline
 import scipy.io.wavfile
@@ -7,22 +7,17 @@ import os
 # Sicherstellen, dass der 'static'-Ordner existiert
 os.makedirs('static', exist_ok=True)
 
-# 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()
-
 # 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:
+        # MusicGen-Pipeline initialisieren (nur innerhalb der GPU-Session)
+        synthesizer = pipeline(
+            "text-to-audio",
+            "facebook/musicgen-small",
+            model_kwargs={"attn_implementation": "eager"}
+        )
+        
         # Musik generieren
         music = synthesizer(prompt, forward_params={"do_sample": True})