artificiallover0 commited on
Commit
d8d52ea
·
verified ·
1 Parent(s): 6e1d375

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -1,8 +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
7
  import torch
8
  from huggingface_hub import login
@@ -27,7 +26,7 @@ MAX_SEED = np.iinfo(np.int32).max
27
  MAX_IMAGE_SIZE = 1024
28
 
29
 
30
- # @spaces.GPU #[uncomment to use ZeroGPU]
31
  def infer(
32
  prompt,
33
  negative_prompt,
 
1
  import gradio as gr
2
  import numpy as np
3
  import random
4
+ import spaces #[uncomment to use ZeroGPU]
 
5
  from diffusers import DiffusionPipeline
6
  import torch
7
  from huggingface_hub import login
 
26
  MAX_IMAGE_SIZE = 1024
27
 
28
 
29
+ @spaces.GPU #[uncomment to use ZeroGPU]
30
  def infer(
31
  prompt,
32
  negative_prompt,