Update app.py

app.py

@@ -1,6 +1,6 @@
 import os
 import pathlib
-import spaces  # Added for ZeroGPU
+import spaces  # Required for ZeroGPU
 import gradio as gr
 import torch
 from PIL import Image
@@ -36,12 +36,13 @@ def update_style_image(style_name: str) -> dict:
     text = get_style_image_markdown_text(style_name)
     return gr.Markdown.update(value=text)
 
-@spaces.GPU(duration=120)  # Added for ZeroGPU, set duration for long inference
+@spaces.GPU(duration=300)  # Increased duration for animation (5 minutes)
 def inference(img, vid):
     if not os.path.exists('temp'):
         os.system('mkdir temp')
 
     img.save("temp/image.jpg", "JPEG")
+    # Check CUDA availability before running
     if torch.cuda.is_available():
         os.system(f"python demo.py --config config/vox-256.yaml --checkpoint ./checkpoints/vox.pth.tar --source_image 'temp/image.jpg' --driving_video {vid} --result_video './temp/result.mp4'")
     else:
@@ -86,7 +87,7 @@ def main():
 
         with gr.Box():
             gr.Markdown('''## Step 3 (Generate Animated Image based on the Video)
-- Hit the **Generate** button. (Note: On cpu-basic, it takes ~ 10 minutes to generate final results.)
+- Hit the **Generate** button. (Note: It may take a few minutes to generate results.)
 ''')
             with gr.Row():
                 with gr.Column():