fix version

app.py

@@ -1,7 +1,10 @@
+
+import spaces
 import os
 import gradio as gr
 import json
 import torch
+
 from huggingface_hub import snapshot_download
 
 from inference_bokehK import load_models as load_bokeh_models, run_inference as run_bokeh_inference, OmegaConf
@@ -37,7 +40,7 @@ shutter_pipeline, shutter_device = load_shutter_models(shutter_cfg)
 color_cfg = OmegaConf.load("configs/inference_genphoto/adv3_256_384_genphoto_relora_color_temperature.yaml")
 color_pipeline, color_device = load_color_models(color_cfg)
 
-
+@spaces.GPU(duration=30)
 def generate_bokeh_video(base_scene, bokehK_list):
     try:
         torch.manual_seed(42) 
@@ -51,6 +54,7 @@ def generate_bokeh_video(base_scene, bokehK_list):
     except Exception as e:
         return f"Error: {str(e)}"
 
+@spaces.GPU(duration=30)
 def generate_focal_video(base_scene, focal_length_list):
     try:
         torch.manual_seed(42)
@@ -64,6 +68,7 @@ def generate_focal_video(base_scene, focal_length_list):
     except Exception as e:
         return f"Error: {str(e)}"
 
+@spaces.GPU(duration=30)
 def generate_shutter_video(base_scene, shutter_speed_list):
     try:
         torch.manual_seed(42)
@@ -77,6 +82,8 @@ def generate_shutter_video(base_scene, shutter_speed_list):
     except Exception as e:
         return f"Error: {str(e)}"
 
+
+@spaces.GPU(duration=30)
 def generate_color_video(base_scene, color_temperature_list):
     try:
         torch.manual_seed(42)

requirements.txt

@@ -1,13 +1,13 @@
---extra-index-url https://download.pytorch.org/whl/cu121
 torch==2.1.1
 torchvision==0.16.1
 torchaudio==2.1.1
-diffusers==0.23.1
+diffusers==0.24.0
 imageio==2.36.0
 imageio-ffmpeg
 numpy==1.24.4
 transformers==4.39.3
 accelerate==0.30.0
+huggingface_hub==0.25.1
 opencv-python
 gdown
 einops
@@ -17,4 +17,4 @@ safetensors
 gradio
 wandb
 triton
-huggingface_hub==0.20.2
+spaces