use ZeroGPU when available

app_wip.py

@@ -1,3 +1,5 @@
+
+import spaces
 import os
 import sys
 import uuid
@@ -195,7 +197,7 @@ def reward_forcing_inference(
     logs += "[WARN] No video generated.\n"
     return None, logs
 
-
+@spaces.GPU(duration=200)
 def gradio_generate(
     prompt: str, duration: str, use_ema: bool, progress=gr.Progress(track_tqdm=True)
 ):