Futuretop commited on
Commit
7e338ff
·
verified ·
1 Parent(s): bd40c11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -60,13 +60,15 @@ def infer(
60
  num_inference_steps=8,
61
  progress=gr.Progress(track_tqdm=True),
62
  ):
 
 
63
  if randomize_seed:
64
  seed = random.randint(0, MAX_SEED)
65
 
66
  generator = torch.Generator().manual_seed(seed)
67
 
68
  image = pipe(
69
- prompt=prompt,
70
  negative_prompt=negative_prompt,
71
  guidance_scale=guidance_scale,
72
  num_inference_steps=num_inference_steps,
@@ -197,7 +199,7 @@ with gr.Blocks(theme="apriel", css=css) as demo:
197
  triggers=[run_button.click, prompt.submit],
198
  fn=infer,
199
  inputs=[
200
- "cartoon styled korean" + prompt,
201
  negative_prompt,
202
  seed,
203
  randomize_seed,
 
60
  num_inference_steps=8,
61
  progress=gr.Progress(track_tqdm=True),
62
  ):
63
+ full_prompt = "cartoon styled korean " + prompt
64
+
65
  if randomize_seed:
66
  seed = random.randint(0, MAX_SEED)
67
 
68
  generator = torch.Generator().manual_seed(seed)
69
 
70
  image = pipe(
71
+ prompt=full_prompt,
72
  negative_prompt=negative_prompt,
73
  guidance_scale=guidance_scale,
74
  num_inference_steps=num_inference_steps,
 
199
  triggers=[run_button.click, prompt.submit],
200
  fn=infer,
201
  inputs=[
202
+ prompt,
203
  negative_prompt,
204
  seed,
205
  randomize_seed,