Koi953215 commited on
Commit
eb5b3bb
1 Parent(s): d4f52fc

update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -115,7 +115,7 @@ def images_to_video(image_list, output_path, fps=10):
115
  writer.close()
116
 
117
 
118
- @spaces.GPU
119
  def NaRCan_make_video(edit_canonical, pth_path, frames_path):
120
  # load NaRCan model
121
  checkpoint_g_old = torch.load(os.path.join(pth_path, "homography_g.pth"))
@@ -193,7 +193,7 @@ def NaRCan_make_video(edit_canonical, pth_path, frames_path):
193
  return edit_video_path
194
 
195
 
196
- @spaces.GPU
197
  def edit_with_pnp(input_video, prompt, num_steps, guidance_scale, seed, n_prompt, control_type="Lineart"):
198
  video_name = input_video.split('/')[-1]
199
  if video_name in video_to_image:
 
115
  writer.close()
116
 
117
 
118
+ @spaces.GPU(duration=120)
119
  def NaRCan_make_video(edit_canonical, pth_path, frames_path):
120
  # load NaRCan model
121
  checkpoint_g_old = torch.load(os.path.join(pth_path, "homography_g.pth"))
 
193
  return edit_video_path
194
 
195
 
196
+ @spaces.GPU(duration=120)
197
  def edit_with_pnp(input_video, prompt, num_steps, guidance_scale, seed, n_prompt, control_type="Lineart"):
198
  video_name = input_video.split('/')[-1]
199
  if video_name in video_to_image: