cwhuh commited on
Commit
57a34a9
·
1 Parent(s): 57fc112

chore : remove prompt refinement

Browse files
Files changed (1) hide show
  1. app.py +8 -9
app.py CHANGED
@@ -46,13 +46,13 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
46
  seed = random.randint(0, MAX_SEED)
47
  generator = torch.Generator().manual_seed(seed)
48
 
49
- refined_prompt = run_gemini(
50
- target_prompt=prompt,
51
- prompt_in_path="prompt.json",
52
- )
53
 
54
  for img in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
55
- prompt=refined_prompt,
56
  guidance_scale=guidance_scale,
57
  num_inference_steps=num_inference_steps,
58
  width=width,
@@ -61,7 +61,7 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
61
  output_type="pil",
62
  good_vae=good_vae,
63
  ):
64
- yield img, seed, refined_prompt
65
 
66
  examples = [
67
  "기계공학과(로켓) 포닉스",
@@ -96,7 +96,6 @@ with gr.Blocks(css=css) as demo:
96
  run_button = gr.Button("Run", scale=0)
97
 
98
  result = gr.Image(label="Result", show_label=False)
99
- refined_prompt_display = gr.Text(label="정제된 프롬프트", show_label=True)
100
 
101
  with gr.Accordion("Advanced Settings", open=False):
102
 
@@ -150,7 +149,7 @@ with gr.Blocks(css=css) as demo:
150
  examples = examples,
151
  fn = infer,
152
  inputs = [prompt],
153
- outputs = [result, seed, refined_prompt_display],
154
  cache_examples="lazy"
155
  )
156
 
@@ -158,7 +157,7 @@ with gr.Blocks(css=css) as demo:
158
  triggers=[run_button.click, prompt.submit],
159
  fn = infer,
160
  inputs = [prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
161
- outputs = [result, seed, refined_prompt_display]
162
  )
163
 
164
  demo.launch()
 
46
  seed = random.randint(0, MAX_SEED)
47
  generator = torch.Generator().manual_seed(seed)
48
 
49
+ # refined_prompt = run_gemini(
50
+ # target_prompt=prompt,
51
+ # prompt_in_path="prompt.json",
52
+ # )
53
 
54
  for img in pipe.flux_pipe_call_that_returns_an_iterable_of_images(
55
+ prompt=prompt,
56
  guidance_scale=guidance_scale,
57
  num_inference_steps=num_inference_steps,
58
  width=width,
 
61
  output_type="pil",
62
  good_vae=good_vae,
63
  ):
64
+ yield img, seed
65
 
66
  examples = [
67
  "기계공학과(로켓) 포닉스",
 
96
  run_button = gr.Button("Run", scale=0)
97
 
98
  result = gr.Image(label="Result", show_label=False)
 
99
 
100
  with gr.Accordion("Advanced Settings", open=False):
101
 
 
149
  examples = examples,
150
  fn = infer,
151
  inputs = [prompt],
152
+ outputs = [result, seed],
153
  cache_examples="lazy"
154
  )
155
 
 
157
  triggers=[run_button.click, prompt.submit],
158
  fn = infer,
159
  inputs = [prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
160
+ outputs = [result, seed]
161
  )
162
 
163
  demo.launch()