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)
|