wjs0725 commited on
Commit
dd393eb
1 Parent(s): 08b018a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -170,7 +170,7 @@ def edit(init_image, source_prompt, target_prompt, num_steps, inject_step, guida
170
  def create_demo(model_name: str, device: str = "cuda:0" if torch.cuda.is_available() else "cpu", offload: bool = False):
171
  is_schnell = model_name == "flux-schnell"
172
  title = r"""
173
- <h1 align="center">Taming Rectified Flow for Inversion and Editing</h1>
174
  """
175
 
176
  description = r"""
@@ -193,7 +193,7 @@ def create_demo(model_name: str, device: str = "cuda:0" if torch.cuda.is_availab
193
  with gr.Blocks(css=css) as demo:
194
  # gr.Markdown(f"# Official Demo for Taming Rectified Flow for Inversion and Editing")
195
 
196
- gr.Markdown(title)
197
  gr.Markdown(description)
198
  gr.Markdown(article)
199
 
 
170
  def create_demo(model_name: str, device: str = "cuda:0" if torch.cuda.is_available() else "cpu", offload: bool = False):
171
  is_schnell = model_name == "flux-schnell"
172
  title = r"""
173
+ <h1 align="center">🪄 Taming Rectified Flow for Inversion and Editing</h1>
174
  """
175
 
176
  description = r"""
 
193
  with gr.Blocks(css=css) as demo:
194
  # gr.Markdown(f"# Official Demo for Taming Rectified Flow for Inversion and Editing")
195
 
196
+ gr.HTML(title)
197
  gr.Markdown(description)
198
  gr.Markdown(article)
199