Nick088 commited on
Commit
eb1a98b
1 Parent(s): 878f8c1

Update infer.py

Browse files
Files changed (1) hide show
  1. infer.py +1 -1
infer.py CHANGED
@@ -28,7 +28,7 @@ def infer_image(img: Image.Image, size_modifier: int ) -> Image.Image:
28
  print(f"Image size ({device}): {size_modifier} ... OK")
29
  return result
30
 
31
- @spaces.GPU(duration=270)
32
  def infer_video(video_filepath: str, size_modifier: int) -> str:
33
  model = RealESRGAN(device, scale=size_modifier)
34
  model.load_weights(f'weights/RealESRGAN_x{size_modifier}.pth', download=False)
 
28
  print(f"Image size ({device}): {size_modifier} ... OK")
29
  return result
30
 
31
+ @spaces.GPU(duration=300)
32
  def infer_video(video_filepath: str, size_modifier: int) -> str:
33
  model = RealESRGAN(device, scale=size_modifier)
34
  model.load_weights(f'weights/RealESRGAN_x{size_modifier}.pth', download=False)