Update app.py

app.py

@@ -10,6 +10,7 @@ from typing import Any, Dict, List, Optional, Union
 import torch
 from PIL import Image
 import gradio as gr
+import spaces
 
 from diffusers import (
     DiffusionPipeline,
@@ -27,8 +28,6 @@ from huggingface_hub import (
 
 from diffusers.utils import load_image
 
-import spaces
-
 def calculate_shift(
     image_seq_len,
     base_seq_len: int = 256,
@@ -2147,7 +2146,7 @@ def update_selection(evt: gr.SelectData, width, height):
         height,
     )
 
-@spaces.GPU(duration=100)
+@spaces.GPU(duration=99)
 def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scale, progress):
     pipe.to("cuda")
     generator = torch.Generator(device="cuda").manual_seed(seed)