Update app.py

app.py

@@ -14,6 +14,7 @@ from diffusers.models.attention_processor import Attention
 from transformers import AutoProcessor, AutoModelForMaskGeneration, pipeline
 from dataclasses import dataclass
 from typing import Any, List, Dict, Optional, Union, Tuple
+import random 
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -208,7 +209,7 @@ print("--- All models loaded successfully! ---")
 
 
 # --- Main Inference Function for Gradio ---
-@spaces.GPU(duration=70)
+@spaces.GPU(duration=85)
 def run_diptych_prompting(
     input_image: Image.Image,
     subject_name: str,