cwhuh commited on
Commit
5250af5
·
1 Parent(s): 4474b1e

add : refinement logic

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -41,10 +41,10 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
41
  seed = random.randint(0, MAX_SEED)
42
  generator = torch.Generator().manual_seed(seed)
43
 
44
- # refined_prompt = run_gemini(
45
- # target_prompt=prompt,
46
- # prompt_in_path="prompt.json",
47
- # )
48
 
49
  for img in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
50
  prompt=prompt,
 
41
  seed = random.randint(0, MAX_SEED)
42
  generator = torch.Generator().manual_seed(seed)
43
 
44
+ refined_prompt = run_gemini(
45
+ target_prompt=prompt,
46
+ prompt_in_path="prompt.json",
47
+ )
48
 
49
  for img in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
50
  prompt=prompt,