Update app.py

app.py

@@ -165,19 +165,19 @@ GPU_DURATION_OPTIONS = {
 def set_gpu_duration(duration_choice):
     os.environ["GPU_DURATION"] = str(GPU_DURATION_OPTIONS[duration_choice])
 
-@gpu_with_duration(duration=45)
+@spaces.GPU(duration=45)
 def generate_45s(*args, **kwargs):
     return generate(*args, **kwargs)  # Call the common generate function
 
-@gpu_with_duration(duration=60)
+@spaces.GPU(duration=60)
 def generate_60s(*args, **kwargs):
     return generate(*args, **kwargs)
 
-@gpu_with_duration(duration=80)
+@spaces.GPU(duration=80)
 def generate_80s(*args, **kwargs):
     return generate(*args, **kwargs)
 
-@gpu_with_duration(duration=100)
+@spaces.GPU(duration=100)
 def generate_100s(*args, **kwargs):
     return generate(*args, **kwargs)