gabboud commited on
Commit
e46d490
·
1 Parent(s): 4eb137b

decorate unconditional_generation function

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -86,6 +86,7 @@ with gr.Blocks(title="RFD3 Test") as demo:
86
 
87
 
88
  # Initialize engine and run generation
 
89
  def unconditional_generation(num_batches, num_designs_per_batch, length):
90
 
91
  config = RFD3InferenceConfig(
 
86
 
87
 
88
  # Initialize engine and run generation
89
+ @spaces.GPU(duration=300)
90
  def unconditional_generation(num_batches, num_designs_per_batch, length):
91
 
92
  config = RFD3InferenceConfig(