xogaurav commited on
Commit
ed51772
β€’
1 Parent(s): d4830a7
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ class FluxGenerator:
40
  flux_generator = FluxGenerator()
41
 
42
 
43
- @spaces.GPU(duration=1)
44
  @torch.inference_mode()
45
  def generate_image(
46
  prompt,
 
40
  flux_generator = FluxGenerator()
41
 
42
 
43
+ @spaces.GPU
44
  @torch.inference_mode()
45
  def generate_image(
46
  prompt,