Disable unnecessary ZeroGPU initialization when not rewriting prompt

app.py

@@ -100,6 +100,18 @@ def set_random_seed(seed: int, deterministic: bool = True) -> None:
 
 
 @spaces.GPU
+def run_rewrite_prompt_gpu(seed, prompt, image_list, progress):
+    set_random_seed(seed)
+    progress(0.05, desc="Rewriting prompt...")
+    if image_list is None:
+        final_prompt = prompt_rewriter.rewrite_text_batch(
+            [prompt])[0]
+    else:
+        final_prompt = prompt_rewriter.rewrite_edit_batch(
+            [image_list], [prompt])[0]
+    return final_prompt
+
+
 def run_rewrite_prompt(seed, prompt, rewrite_prompt, in_image, progress=gr.Progress(track_tqdm=True)):
     image_list = None
     if in_image is not None and len(in_image) > 0:
@@ -107,14 +119,7 @@ def run_rewrite_prompt(seed, prompt, rewrite_prompt, in_image, progress=gr.Progr
         for item in in_image:
             image_list.append(item[0])
     if rewrite_prompt:
-        set_random_seed(seed)
-        progress(0.05, desc="Rewriting prompt...")
-        if image_list is None:
-            final_prompt = prompt_rewriter.rewrite_text_batch(
-                [prompt])[0]
-        else:
-            final_prompt = prompt_rewriter.rewrite_edit_batch(
-                [image_list], [prompt])[0]
+        final_prompt = run_rewrite_prompt_gpu(seed, prompt, image_list, progress)
         return final_prompt, None
     else:
         return '', None