cbensimon HF staff commited on
Commit
fcaf80d
1 Parent(s): 93bb24a

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 = functools.partial(process_image, pipe)
283
- process_pipe_video = functools.partial(process_video, pipe)
284
- process_pipe_bas = 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(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()