Update app.py

app.py

@@ -1,16 +1,18 @@
 import gradio as gr
 import os
 from model import Jamify
-
+import spaces
 # Initialize the Jamify model once
 print("Initializing Jamify model...")
 jamify_model = Jamify()
 print("Jamify model ready.")
 
+
+@spaces.GPU(duration=100)
 def generate_song(reference_audio, lyrics_file, style_prompt, duration):
     # We need to save the uploaded files to temporary paths to pass to the model
-    ref_audio_path = reference_audio.name if reference_audio else None
-    lyrics_path = lyrics_file.name
+    reference_audio = reference_audio if reference_audio else None
+    
 
     # The model expects paths, so we write the prompt to a temp file if needed
     # (This part of the model could be improved to accept the string directly)