zhangyang-0123 commited on
Commit
ab5ec93
•
1 Parent(s): 3735327

update header

Browse files
Files changed (1) hide show
  1. app.py +17 -9
app.py CHANGED
@@ -87,22 +87,30 @@ def on_generate_click(prompt, seed, steps, pipe, pruned_pipe):
87
 
88
  header = """
89
  # 🌱 Text-to-Image Generation with EcoDiff Pruned Models
90
-
91
- ![Static Badge](https://img.shields.io/badge/ariXv-Paper-A42C25?link=https://arxiv.org/abs/2412.02852)
92
- ![Static Badge](https://img.shields.io/badge/🤗-Model-ffbd45?link=https://huggingface.co/zhangyang-0123/EcoDiffPrunedModels)
93
- ![Static Badge](https://img.shields.io/badge/GitHub-Code-blue?logo=github&link=https://github.com/YaNgZhAnG-V5/EcoDiff)
94
  """
95
  header_2 = """
96
- For âš¡ <b>faster</b> âš¡ DEMO on one model only, please visit
97
- ![Static Badge](https://img.shields.io/badge/SDXL-fedcba?link=https://huggingface.co/spaces/zhangyang-0123/EcoDiff-SD-XL)
98
- ![Static Badge](https://img.shields.io/badge/FLUX-fgdfba?link=https://huggingface.co/spaces/zhangyang-0123/EcoDiff-FLUX-Schnell)
 
 
 
 
 
 
 
 
99
  """
100
 
101
 
102
  def create_demo():
103
  with gr.Blocks() as demo:
104
- gr.Markdown(header)
105
- gr.Markdown(header_2)
 
 
 
 
106
  with gr.Row():
107
  gr.Markdown(
108
  """
 
87
 
88
  header = """
89
  # 🌱 Text-to-Image Generation with EcoDiff Pruned Models
 
 
 
 
90
  """
91
  header_2 = """
92
+ <div style="text-align: center; display: flex; justify-content: left; gap: 5px;">
93
+ <a href="https://arxiv.org/abs/2412.02852"><img src="https://img.shields.io/badge/ariXv-Paper-A42C25.svg" alt="arXiv"></a>
94
+ <a href="https://huggingface.co/zhangyang-0123/EcoDiffPrunedModels"><img src="https://img.shields.io/badge/🤗-Model-ffbd45.svg" alt="HuggingFace"></a>
95
+ </div>
96
+ """
97
+ header_3 = """
98
+ <div style="text-align: center; display: flex; justify-content: left; gap: 5px;">
99
+ For âš¡ <b>faster</b> âš¡ DEMO on one model only, please visit
100
+ <a href="https://huggingface.co/spaces/zhangyang-0123/EcoDiff-SD-XL"><img alt="Static Badge" src="https://img.shields.io/badge/SDXL-fedcba.svg"></a>
101
+ <a href="https://huggingface.co/spaces/zhangyang-0123/EcoDiff-FLUX-Schnell"><img alt="Static Badge" src="https://img.shields.io/badge/FLUX-fgdfba"></a>
102
+ </div>
103
  """
104
 
105
 
106
  def create_demo():
107
  with gr.Blocks() as demo:
108
+ with gr.Row():
109
+ gr.Markdown(header)
110
+ with gr.Row():
111
+ gr.HTML(header_2)
112
+ with gr.Row():
113
+ gr.HTML(header_3)
114
  with gr.Row():
115
  gr.Markdown(
116
  """