haodongli commited on
Commit
6fcc783
1 Parent(s): 25fc9d6

update for zerogpu

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -42,8 +42,8 @@ def infer_video(path_input, seed):
42
  return [g_save_path, d_save_path]
43
 
44
  def run_demo_server():
45
- infer_gpu = spaces.GPU(functools.partial(infer, seed=0))
46
- infer_video_gpu = spaces.GPU(functools.partial(infer_video, seed=0))
47
  gradio_theme = gr.themes.Default()
48
 
49
  with gr.Blocks(
 
42
  return [g_save_path, d_save_path]
43
 
44
  def run_demo_server():
45
+ infer_gpu = spaces.GPU(functools.partial(infer))
46
+ infer_video_gpu = spaces.GPU(functools.partial(infer_video))
47
  gradio_theme = gr.themes.Default()
48
 
49
  with gr.Blocks(