Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -20,8 +20,11 @@ MAX_IMAGE_SIZE = 2048
|
|
20 |
|
21 |
pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe)
|
22 |
|
23 |
-
@spaces.GPU(duration=
|
24 |
-
def infer(prompt, seed=42, randomize_seed=
|
|
|
|
|
|
|
25 |
if randomize_seed:
|
26 |
seed = random.randint(0, MAX_SEED)
|
27 |
generator = torch.Generator().manual_seed(seed)
|
@@ -136,4 +139,4 @@ FLUX.1 Krea [dev] model further tuned and customized with [Krea](https://krea.ai
|
|
136 |
outputs = [result, seed]
|
137 |
)
|
138 |
|
139 |
-
demo.launch()
|
|
|
20 |
|
21 |
pipe.flux_pipe_call_that_returns_an_iterable_of_images = flux_pipe_call_that_returns_an_iterable_of_images.__get__(pipe)
|
22 |
|
23 |
+
@spaces.GPU(duration=25)
|
24 |
+
def infer(prompt, seed=42, randomize_seed=True, width=768, height=768, guidance_scale=4.5, num_inference_steps=20, progress=gr.Progress(track_tqdm=True)):
|
25 |
+
"""
|
26 |
+
Generate an image using the Flux.1 Krea-Dev Image Generator
|
27 |
+
"""
|
28 |
if randomize_seed:
|
29 |
seed = random.randint(0, MAX_SEED)
|
30 |
generator = torch.Generator().manual_seed(seed)
|
|
|
139 |
outputs = [result, seed]
|
140 |
)
|
141 |
|
142 |
+
demo.launch(mcp_server=True)
|