crevelop commited on
Commit
d9353e3
·
unverified ·
1 Parent(s): 05de3c4

use ZeroGPU spaces

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import gradio as gr
2
  import numpy as np
3
  import random
@@ -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 spaces
2
  import gradio as gr
3
  import numpy as np
4
  import random
 
22
  MAX_IMAGE_SIZE = 1024
23
 
24
 
25
+ @spaces.GPU
26
  def infer(
27
  prompt,
28
  negative_prompt,