Hugging Face's logo

Spaces:
fffiloni
/
svd_keyframe_interpolation
Running on Zero

App Files Community
fffiloni commited on
Commit
4cb6a68
·
verified ·
1 Parent(s): 2277572

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. gradio_app.py +2 -2
gradio_app.py CHANGED
@@ -85,7 +85,7 @@ def cuda_memory_cleanup():
85
  torch.cuda.ipc_collect()
86
  gc.collect()
87
 
88
- @spaces.GPU()
89
  def infer(frame1_path, frame2_path):
90
 
91
  seed = 42
@@ -113,7 +113,7 @@ def infer(frame1_path, frame2_path):
113
  weighted_average=weighted_average, # True
114
  noise_injection_steps=noise_injection_steps, # 0
115
  noise_injection_ratio= noise_injection_ratio, # 0.5
116
- decode_chunk_size=6
117
  ).frames[0]
118
 
119
  cuda_memory_cleanup()
 
85
  torch.cuda.ipc_collect()
86
  gc.collect()
87
 
88
+ @spaces.GPU(duration=90)
89
  def infer(frame1_path, frame2_path):
90
 
91
  seed = 42
 
113
  weighted_average=weighted_average, # True
114
  noise_injection_steps=noise_injection_steps, # 0
115
  noise_injection_ratio= noise_injection_ratio, # 0.5
116
+ #decode_chunk_size=6
117
  ).frames[0]
118
 
119
  cuda_memory_cleanup()