gokaygokay commited on
Commit
e538d2e
Β·
verified Β·
1 Parent(s): c0b5afb

Update pixelize_processor.py

Browse files
Files changed (1) hide show
  1. pixelize_processor.py +2 -0
pixelize_processor.py CHANGED
@@ -1,8 +1,10 @@
 
1
  import numpy as np
2
  from pixeloe.pixelize import pixelize
3
  from PIL import Image
4
  import gradio as gr
5
 
 
6
  def pixelize_image(
7
  image,
8
  downscale_mode="contrast",
 
1
+ import spaces
2
  import numpy as np
3
  from pixeloe.pixelize import pixelize
4
  from PIL import Image
5
  import gradio as gr
6
 
7
+ @spaces.GPU
8
  def pixelize_image(
9
  image,
10
  downscale_mode="contrast",