Update app.py

app.py

@@ -184,10 +184,35 @@ def preprocess_n_magnify(input_image: Image.Image):
     preprocessed_img, magnified_img = magnify(preprocessed_img)
 
     return preprocessed_img, (preprocessed_img, magnified_img)
+
+def get_duration(
+    input_image: Image.Image,
+    duration_seconds,
+    user_prompt,
+    positive_prompt,
+    negative_prompt,
+    num_inference_steps,
+    scale_factor,
+    cfg_scale,
+    seed,
+    latent_tiled_size,
+    latent_tiled_overlap,
+    sample_times,
+    progress,
+    ):
     
-@spaces.GPU()
+    if steps > 4 and duration_seconds > 2:
+        return 90
+    elif steps > 4 or duration_seconds > 2:
+        return 75
+    else:
+        return 60
+        
+    
+@spaces.GPU(duration=get_duration)
 def magnify(
     input_image: Image.Image,
+    duration_seconds = 1,
     user_prompt = "",
     positive_prompt = "clean, high-resolution, 8k, best quality, masterpiece",
     negative_prompt = "dotted, noise, blur, lowres, oversmooth, longbody, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality",
@@ -199,7 +224,7 @@ def magnify(
     latent_tiled_overlap = 4,
     sample_times = 1,
     progress=gr.Progress(track_tqdm=True),
-    ) -> List[np.ndarray]:
+    ):