ailm commited on
Commit
b53b1e0
·
verified ·
1 Parent(s): 8574cae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ from dataTransform import tensor_to_image
17
  device = 'cuda'
18
  print(device)
19
 
20
- @spaces.GPU(duration = 500)
21
  def gradio_style_transfer(steps, content_image, style_image):
22
  generated_tensor = style_transfer(content_image, style_image, total_steps= steps)
23
  generated_image = tensor_to_image(generated_tensor)
 
17
  device = 'cuda'
18
  print(device)
19
 
20
+ @spaces.GPU(duration = 242)
21
  def gradio_style_transfer(steps, content_image, style_image):
22
  generated_tensor = style_transfer(content_image, style_image, total_steps= steps)
23
  generated_image = tensor_to_image(generated_tensor)