Spaces:
Running
on
Zero
Running
on
Zero
add : refinement logic
Browse files
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 |
-
|
45 |
-
|
46 |
-
|
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,
|