Update

app.py

@@ -100,7 +100,7 @@ def rollout(attentions, discard_ratio, head_fusion, num_prefix_tokens=1):
     mask = mask / np.max(mask)
     return mask
 
-@spaces.GPU(duration=120)  # Allowing up to 120 seconds for this operation
+@spaces.GPU
 def visualize_attention(
         image: Image.Image,
         model_name: str,