launch-calcium commited on
Commit
de3db19
·
verified ·
1 Parent(s): f7e25bd

Update app.py

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