unicodeveloper commited on
Commit
0973562
·
verified ·
1 Parent(s): 81caa67

add spaces

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,6 +1,7 @@
1
  import gradio as gr
2
  import numpy as np
3
  import random
 
4
 
5
  # import spaces #[uncomment to use ZeroGPU]
6
  from diffusers import DiffusionPipeline
@@ -21,7 +22,7 @@ MAX_SEED = np.iinfo(np.int32).max
21
  MAX_IMAGE_SIZE = 1024
22
 
23
 
24
- # @spaces.GPU #[uncomment to use ZeroGPU]
25
  def infer(
26
  prompt,
27
  negative_prompt,
 
1
  import gradio as gr
2
  import numpy as np
3
  import random
4
+ import spaces
5
 
6
  # import spaces #[uncomment to use ZeroGPU]
7
  from diffusers import DiffusionPipeline
 
22
  MAX_IMAGE_SIZE = 1024
23
 
24
 
25
+ @spaces.GPU
26
  def infer(
27
  prompt,
28
  negative_prompt,