Spaces:

RED-AIGC
/

TDD

Sleeping

BILLY12138 commited on Sep 3, 2024

Commit

fc3d45b

verified ·

1 Parent(s): 95f0824

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,4 +1,4 @@
-#import spaces
 import gradio as gr
 import torch
 from diffusers import StableDiffusionXLPipeline, UNet2DConditionModel
@@ -79,7 +79,7 @@ if SAFETY_CHECKER:
         return images, has_nsfw_concepts
-#@spaces.GPU(enable_queue=True)
 def generate_image(
     prompt,
     negative_prompt,

+import spaces
 import gradio as gr
 import torch
 from diffusers import StableDiffusionXLPipeline, UNet2DConditionModel
         return images, has_nsfw_concepts
+@spaces.GPU(enable_queue=True)
 def generate_image(
     prompt,
     negative_prompt,