radames commited on
Commit
ba4e386
1 Parent(s): 85c2929

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -8,6 +8,7 @@ import requests
8
  import io
9
  import os
10
  from PIL import Image
 
11
 
12
  from diffusers import (
13
  StableDiffusionPipeline,
@@ -61,7 +62,7 @@ SAMPLER_MAP = {
61
  "DEIS": lambda config: DEISMultistepScheduler.from_config(config),
62
  }
63
 
64
-
65
  def inference(
66
  qr_code_content: str,
67
  prompt: str,
 
8
  import io
9
  import os
10
  from PIL import Image
11
+ import spaces
12
 
13
  from diffusers import (
14
  StableDiffusionPipeline,
 
62
  "DEIS": lambda config: DEISMultistepScheduler.from_config(config),
63
  }
64
 
65
+ @spaces.GPU()
66
  def inference(
67
  qr_code_content: str,
68
  prompt: str,