rphrp1985 commited on
Commit
ff7ba90
1 Parent(s): 51eaf64

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,7 +15,7 @@ is_canonical = True if os.environ.get("SPACE_ID") == "Pyramid-Flow/pyramid-flow"
15
  # Constants
16
  MODEL_PATH = "pyramid-flow-model"
17
  MODEL_REPO = "rain1011/pyramid-flow-sd3"
18
- MODEL_VARIANT = "diffusion_transformer_768p"
19
  MODEL_DTYPE = "bf16"
20
 
21
  def center_crop(image, target_width, target_height):
@@ -95,7 +95,7 @@ def generate_video(prompt, image=None, duration=10, guidance_scale=9, video_guid
95
  save_memory=True,
96
  )
97
  output_path = f"{str(uuid.uuid4())}_output_video.mp4"
98
- export_to_video(frames, output_path, fps=frames_per_second)
99
  return output_path
100
 
101
  # Gradio interface
 
15
  # Constants
16
  MODEL_PATH = "pyramid-flow-model"
17
  MODEL_REPO = "rain1011/pyramid-flow-sd3"
18
+ MODEL_VARIANT = "diffusion_transformer_384p"
19
  MODEL_DTYPE = "bf16"
20
 
21
  def center_crop(image, target_width, target_height):
 
95
  save_memory=True,
96
  )
97
  output_path = f"{str(uuid.uuid4())}_output_video.mp4"
98
+ export_to_video(frames, output_path, fps= 8)
99
  return output_path
100
 
101
  # Gradio interface