cbensimon HF staff commited on
Commit
6824586
1 Parent(s): 9fff649

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -279,9 +279,9 @@ def process_bas(
279
 
280
 
281
  def run_demo_server(pipe):
282
- process_pipe_image = spaces.GPU(functools.partial(process_image, pipe))
283
- process_pipe_video = spaces.GPU(functools.partial(process_video, pipe))
284
- process_pipe_bas = spaces.GPU(functools.partial(process_bas, pipe))
285
  os.environ["GRADIO_ALLOW_FLAGGING"] = "never"
286
 
287
  gradio_theme = gr.themes.Default()
 
279
 
280
 
281
  def run_demo_server(pipe):
282
+ process_pipe_image = spaces.GPU(lambda *args, **kwargs: process_image(pipe, *args, **kwargs))
283
+ process_pipe_video = spaces.GPU(lambda *args, **kwargs: process_video(pipe, *args, **kwargs))
284
+ process_pipe_bas = spaces.GPU(lambda *args, **kwargs: process_bas(pipe, *args, **kwargs))
285
  os.environ["GRADIO_ALLOW_FLAGGING"] = "never"
286
 
287
  gradio_theme = gr.themes.Default()