ameerazam08 commited on
Commit
c722062
β€’
1 Parent(s): b57786a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ import gradio as gr
7
  import numpy as np
8
  import PIL.Image
9
 
10
- # import spaces
11
  import torch
12
  from diffusers import (
13
  StableDiffusionXLPipeline,
@@ -65,7 +65,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
65
  return seed
66
 
67
 
68
- # @spaces.GPU(enable_queue=True)
69
  def generate(
70
  prompt: str,
71
  negative_prompt: str = "",
 
7
  import numpy as np
8
  import PIL.Image
9
 
10
+ import spaces
11
  import torch
12
  from diffusers import (
13
  StableDiffusionXLPipeline,
 
65
  return seed
66
 
67
 
68
+ @spaces.GPU(enable_queue=True)
69
  def generate(
70
  prompt: str,
71
  negative_prompt: str = "",