Update app.py

app.py

@@ -131,9 +131,8 @@ def check_duration(
         return 360
     else:
         return 900
-    
 
-@spaces.GPU(check_duration)
+@spaces.GPU(duration=check_duration)
 def generate_video(
     mode,
     prompt,