ameerazam08 commited on
Commit
f8f94e7
1 Parent(s): 064f75e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ import random
8
  import gradio as gr
9
  import numpy as np
10
  import PIL.Image
11
- # import spaces
12
  import torch
13
  from diffusers import AutoPipelineForText2Image, DPMSolverMultistepScheduler
14
  from huggingface_hub import hf_hub_download
@@ -50,7 +50,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
50
  return seed
51
 
52
 
53
- # @spaces.GPU(enable_queue=True)
54
  def generate(
55
  prompt: str,
56
  negative_prompt: str = "",
 
8
  import gradio as gr
9
  import numpy as np
10
  import PIL.Image
11
+ import spaces
12
  import torch
13
  from diffusers import AutoPipelineForText2Image, DPMSolverMultistepScheduler
14
  from huggingface_hub import hf_hub_download
 
50
  return seed
51
 
52
 
53
+ @spaces.GPU(enable_queue=True)
54
  def generate(
55
  prompt: str,
56
  negative_prompt: str = "",