Update app.py

app.py

@@ -47,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,
@@ -66,6 +66,7 @@ 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.
         '''
+    @spaces.GPU(duration=60 * 3)
     def generate_image(
             is_mega: bool,
             text: str,