Update app.py

app.py

@@ -24,7 +24,7 @@ def image_gen(click,rows = 4,cols = 4):
     return grid
     #return 'button clicked'
     
-@spaces.GPU
+@spaces.GPU(duration = 300)
 def image_gen_modified(rows=4,cols=4):
   pic_hold = []
   model_output = pipeline(batch_size=16).images