Update app.py

app.py

@@ -145,7 +145,7 @@ def to_wanted_embs(image_outputs, input_ids, attention_mask, cache_position=None
     
 
 # TODO cache descriptions?
-@spaces.GPU()
+@spaces.GPU(duration=20)
 def generate_pali(n_embs):
     prompt = 'caption en'
     model_inputs = processor(text=prompt, images=torch.zeros(1, 3, 224, 224), return_tensors="pt")
@@ -180,7 +180,7 @@ def generate_pali(n_embs):
 
 
 
-@spaces.GPU()
+@spaces.GPU(duration=20)
 def generate_gpu(in_im_embs, prompt='the scene'):
     with torch.no_grad():
         in_im_embs = in_im_embs.to('cuda').unsqueeze(0).unsqueeze(0)