akhaliq HF staff commited on
Commit
02741dc
1 Parent(s): 0288612

add duplicate spaces badge

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -80,6 +80,9 @@ def generate(image_size, vae_model, class_label, cfg_scale, num_sampling_steps,
80
  description = '''This is a demo of our DiT image generation models. DiTs are a new class of diffusion models with
81
  transformer backbones. They are class-conditional models trained on ImageNet-1K, and they outperform prior DDPMs.'''
82
 
 
 
 
83
  project_links = '''
84
  <p style="text-align: center">
85
  <a href="https://www.wpeebles.com/DiT.html">Project Page</a> &#183;
@@ -104,6 +107,7 @@ with gr.Blocks() as demo:
104
  gr.Markdown("<h1 style='text-align: center'>Scalable Diffusion Models with Transformers (DiT)</h1>")
105
  gr.Markdown(project_links)
106
  gr.Markdown(description)
 
107
 
108
  with gr.Tabs():
109
  with gr.TabItem('Generate'):
 
80
  description = '''This is a demo of our DiT image generation models. DiTs are a new class of diffusion models with
81
  transformer backbones. They are class-conditional models trained on ImageNet-1K, and they outperform prior DDPMs.'''
82
 
83
+ duplicate = '''Skip the queue by duplicating this space and upgrading to GPU in settings
84
+ <a href="https://huggingface.co/spaces/wpeebles/DiT?duplicate=true"><img src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a>'''
85
+
86
  project_links = '''
87
  <p style="text-align: center">
88
  <a href="https://www.wpeebles.com/DiT.html">Project Page</a> &#183;
 
107
  gr.Markdown("<h1 style='text-align: center'>Scalable Diffusion Models with Transformers (DiT)</h1>")
108
  gr.Markdown(project_links)
109
  gr.Markdown(description)
110
+ gr.Markdown(duplicate)
111
 
112
  with gr.Tabs():
113
  with gr.TabItem('Generate'):