cwhuh commited on
Commit
e783b74
·
1 Parent(s): c16a8c1

chore : add log for user prompt

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -42,11 +42,13 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
42
  seed = random.randint(0, MAX_SEED)
43
  generator = torch.Generator().manual_seed(seed)
44
 
 
 
45
  refined_prompt = run_gemini(
46
  target_prompt=prompt,
47
  prompt_in_path="prompt.json",
48
  )
49
- print(f"Refined prompt: {refined_prompt}")
50
 
51
  for img in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
52
  prompt=refined_prompt,
 
42
  seed = random.randint(0, MAX_SEED)
43
  generator = torch.Generator().manual_seed(seed)
44
 
45
+ print(f"User Prompt: {prompt}")
46
+
47
  refined_prompt = run_gemini(
48
  target_prompt=prompt,
49
  prompt_in_path="prompt.json",
50
  )
51
+ print(f"Refined Prompt: {refined_prompt}")
52
 
53
  for img in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
54
  prompt=refined_prompt,