Update model.py

model.py

@@ -88,7 +88,7 @@ class Model:
         return t5_model
 
     @torch.no_grad()
-    @spaces.GPU(enable_queue=True)
+    @spaces.GPU(enable_queue=True, duration=120)
     def process_edge(
         self,
         image: np.ndarray,
@@ -188,7 +188,7 @@ class Model:
         return samples
 
     @torch.no_grad()
-    @spaces.GPU(enable_queue=True)
+    @spaces.GPU(enable_queue=True, duration=120)
     def process_depth(
         self,
         image: np.ndarray,
@@ -237,7 +237,7 @@ class Model:
         for idx, (caption_emb,
                   emb_mask) in enumerate(zip(caption_embs, emb_masks)):
             valid_num = int(emb_mask.sum().item())
-            print(f'  prompt {idx} token len: {valid_num}')
+            print(f'prompt {idx} token len: {valid_num}')
             new_caption_emb = torch.cat(
                 [caption_emb[valid_num:], caption_emb[:valid_num]])
             new_caption_embs.append(new_caption_emb)