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

requirements.txt

@@ -4,4 +4,5 @@ torch
 transformers==4.42.4
 xformers
 sentencepiece
-peft==0.12.0
+peft==0.12.0
+google-genai