test : minor changes

app.py

@@ -7,7 +7,7 @@ from diffusers import  DiffusionPipeline, FlowMatchEulerDiscreteScheduler, Autoe
 from transformers import CLIPTextModel, CLIPTokenizer,T5EncoderModel, T5TokenizerFast
 from live_preview_helpers import calculate_shift, retrieve_timesteps, flux_pipe_call_that_returns_an_iterable_of_images
 
-from llm_wrapper import run_gemini
+# from llm_wrapper import run_gemini
 from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
 import subprocess
@@ -41,10 +41,10 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator().manual_seed(seed)
 
-    refined_prompt = run_gemini(
-        target_prompt=prompt,
-        prompt_in_path="prompt.json",
-    )
+    # refined_prompt = run_gemini(
+    #     target_prompt=prompt,
+    #     prompt_in_path="prompt.json",
+    # )
     
     for img in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
             prompt=refined_prompt,