svjack commited on
Commit
ba50c62
·
verified ·
1 Parent(s): a531730

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -12,7 +12,7 @@ from src.gradio_pipeline import GradioPipeline
12
  from src.config.crop_config import CropConfig
13
  from src.config.argument_config import ArgumentConfig
14
  from src.config.inference_config import InferenceConfig
15
- import spaces
16
  import cv2
17
 
18
  # import gdown
@@ -36,11 +36,11 @@ gradio_pipeline = GradioPipeline(
36
  args=args
37
  )
38
 
39
- @spaces.GPU(duration=240)
40
  def gpu_wrapped_execute_video(*args, **kwargs):
41
  return gradio_pipeline.execute_video(*args, **kwargs)
42
 
43
- @spaces.GPU(duration=240)
44
  def gpu_wrapped_execute_image(*args, **kwargs):
45
  return gradio_pipeline.execute_image(*args, **kwargs)
46
 
 
12
  from src.config.crop_config import CropConfig
13
  from src.config.argument_config import ArgumentConfig
14
  from src.config.inference_config import InferenceConfig
15
+ #import spaces
16
  import cv2
17
 
18
  # import gdown
 
36
  args=args
37
  )
38
 
39
+ #@spaces.GPU(duration=240)
40
  def gpu_wrapped_execute_video(*args, **kwargs):
41
  return gradio_pipeline.execute_video(*args, **kwargs)
42
 
43
+ #@spaces.GPU(duration=240)
44
  def gpu_wrapped_execute_image(*args, **kwargs):
45
  return gradio_pipeline.execute_image(*args, **kwargs)
46