evalstate HF Staff commited on
Commit
e5681ab
·
verified ·
1 Parent(s): c925b1b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
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=75)
24
- def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=4.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
 
 
 
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)