robin-courant commited on
Commit
969214b
1 Parent(s): 1521648

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -140,6 +140,7 @@ def generate(
140
  # ------------------------------------------------------------------------------------- #
141
 
142
 
 
143
  def main(gen_fn: Callable):
144
  theme = gr.themes.Default(primary_hue="blue", secondary_hue="gray")
145
 
 
140
  # ------------------------------------------------------------------------------------- #
141
 
142
 
143
+ @spaces.GPU
144
  def main(gen_fn: Callable):
145
  theme = gr.themes.Default(primary_hue="blue", secondary_hue="gray")
146