feat: zerogpu support

app.py

@@ -16,6 +16,7 @@ from media_pipe.draw_util import FaceMeshVisualizer
 
 from download_models import download 
 import torch
+import spaces  # ZeroGPU support
 
 # Download models and check for exists
 download()
@@ -30,7 +31,7 @@ if torch.cuda.is_available():
     print(f"   GPU device: {torch.cuda.get_device_name(0)}")
     print(f"   GPU count: {torch.cuda.device_count()}")
 else:
-    print("   ⚠️  No CUDA GPU detected - will use CPU")
+    print("   ℹ️  GPU will be allocated on-demand (ZeroGPU mode)")
 print("="*50)
 
 PROCESSED_VIDEO_DIR = './processed_videos'
@@ -184,6 +185,7 @@ def preview_crop(image_path, npy_file, video_path, expand_x, expand_y, offset_x,
     else:
         return None,None, "Failed to generate crop preview"
 
+@spaces.GPU(duration=120)  # Allocate GPU for 120 seconds (adjust based on your video length)
 def generate_video(input_img, should_crop_face, expand_x, expand_y, offset_x, offset_y, input_video_type, input_video, input_npy_select, input_npy, input_video_frames,
                    settings_steps, settings_cfg_scale, settings_seed, resolution_w, resolution_h,
                    model_step, custom_output_path, use_custom_fps, output_fps, callback_steps, context_frames, context_stride, context_overlap, context_batch_size, anomaly_action,intropolate_factor):
@@ -278,9 +280,10 @@ with gr.Blocks() as demo:
         gpu_name = torch.cuda.get_device_name(0)
         gpu_info = f"🚀 **GPU Enabled**: {gpu_name}"
     else:
-        gpu_info = "⚠️ **Running on CPU** (Generation will be slower)"
+        # ZeroGPU mode - GPU allocated on-demand
+        gpu_info = "⚡ **ZeroGPU Mode**: GPU will be allocated automatically when generating"
     
-    gr.Markdown(f"<div style='text-align: center; padding: 10px; background-color: #f0f0f0; border-radius: 5px;'>{gpu_info}</div>")
+    gr.Markdown(f"<div style='text-align: center; padding: 10px; background-color: #e8f5e9; border-radius: 5px; border: 1px solid #4caf50;'>{gpu_info}</div>")
     
     gr.Markdown("""
     <div style='text-align: center; padding: 20px; background-color: #f8f9fa; border-radius: 10px; margin: 10px 0; border: 2px solid #e0e0e0;'>

requirements.txt

@@ -26,3 +26,4 @@ xformers==0.0.23
 scikit-image
 deepspeed==0.13.1
 gradio==4.39.0
+spaces