reedmayhew commited on
Commit
65b549e
1 Parent(s): 337146c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ def upscale_chunk(chunk, model, processor, device):
43
  output_image = (output * 255.0).round().astype(np.uint8)
44
  return Image.fromarray(output_image)
45
 
46
- @spaces.GPU(duration=120)
47
  def main(image, model_choice, save_as_jpg=True):
48
  # Resize the input image
49
  image = resize_image(image)
 
43
  output_image = (output * 255.0).round().astype(np.uint8)
44
  return Image.fromarray(output_image)
45
 
46
+ @spaces.GPU
47
  def main(image, model_choice, save_as_jpg=True):
48
  # Resize the input image
49
  image = resize_image(image)