Update app.py

app.py

@@ -12,7 +12,18 @@ import os
 import glob
 import subprocess
 import imageio_ffmpeg
+import os
 
+# Define a fallback for environments without GPU
+if os.environ.get("SPACES_ZERO_GPU") is not None:
+    import spaces
+else:
+    class spaces:
+        @staticmethod
+        def GPU(func):
+            def wrapper(*args, **kwargs):
+                return func(*args, **kwargs)
+            return wrapper
 # Ensure 'punkt' is downloaded for nltk
 try:
     nltk.data.find('tokenizers/punkt')
@@ -36,7 +47,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 model.to(device)
 print(device)
 
-
+@spaces.GPU(duration=60 * 3)
 def get_output_video(text):
     inputs = tokenizer(text,
                        max_length=1024,
@@ -55,7 +66,6 @@ def get_output_video(text):
         Set the device to either "cuda" or "cpu". Once everything has finished initializing,
     float32 is faster than float16 but uses more GPU memory.
         '''
-
     def generate_image(
             is_mega: bool,
             text: str,