yasserrmd commited on
Commit
73ee197
1 Parent(s): 7341603

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ from huggingface_hub import snapshot_download
7
  from pyramid_dit import PyramidDiTForVideoGeneration
8
  from diffusers.utils import export_to_video
9
 
10
- import spaces
11
  import uuid
12
 
13
  # Constants
@@ -59,7 +59,7 @@ def load_model():
59
  model = load_model()
60
 
61
  # Text-to-video generation function
62
- @spaces.GPU(duration=120)
63
  def generate_video(prompt, image=None, duration=5, guidance_scale=9, video_guidance_scale=5, progress=gr.Progress(track_tqdm=True)):
64
  multiplier = 3
65
  temp = int(duration * multiplier) + 1 # Convert seconds to temp value (assuming 24 FPS)
 
7
  from pyramid_dit import PyramidDiTForVideoGeneration
8
  from diffusers.utils import export_to_video
9
 
10
+ #import spaces
11
  import uuid
12
 
13
  # Constants
 
59
  model = load_model()
60
 
61
  # Text-to-video generation function
62
+ #@spaces.GPU(duration=120)
63
  def generate_video(prompt, image=None, duration=5, guidance_scale=9, video_guidance_scale=5, progress=gr.Progress(track_tqdm=True)):
64
  multiplier = 3
65
  temp = int(duration * multiplier) + 1 # Convert seconds to temp value (assuming 24 FPS)