InspireBake

Sleeping

yasserrmd commited on Jan 4

Commit

33eae09

verified ·

1 Parent(s): eb9a48b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ from transformers import (
 )
 from diffusers import VQModel
 import gradio as gr
-import spaces
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 dtype = torch.bfloat16
@@ -32,7 +32,7 @@ pipe.to(device)
 MAX_SEED = 2**32 - 1
-@spaces.GPU
 def generate_image(occasion, theme, colors, randomize_seed=True, seed=0):
     prompt = f"{occasion} theme: {theme}, colors: {colors} design inspiration"
     if randomize_seed or seed == 0:

 )
 from diffusers import VQModel
 import gradio as gr
+#import spaces
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 dtype = torch.bfloat16
 MAX_SEED = 2**32 - 1
+#@spaces.GPU
 def generate_image(occasion, theme, colors, randomize_seed=True, seed=0):
     prompt = f"{occasion} theme: {theme}, colors: {colors} design inspiration"
     if randomize_seed or seed == 0: