Sreerama commited on
Commit
ce5df75
1 Parent(s): 624f047

add extra step for spaces setting

Browse files
Files changed (1) hide show
  1. app.py +18 -4
app.py CHANGED
@@ -417,10 +417,24 @@ with gr.Blocks(css=css) as demo:
417
  ''')
418
  else:
419
  top_description = gr.HTML(f'''
420
- <div class="gr-prose" style="max-width: 80%">
421
- <h2>You have successfully duplicated the Dreambooth Training Space 🎉</h2>
422
- <p>There's only one step left before you can train your model: <a href="https://huggingface.co/spaces/{os.environ['SPACE_ID']}/settings" style="text-decoration: underline" target="_blank">attribute a <b>T4 GPU</b> to it (via the Settings tab)</a> and run the training below. Other GPUs are not compatible for now. You will be billed by the minute from when you activate the GPU until when it is turned it off.</p>
423
- </div>
 
 
 
 
 
 
 
 
 
 
 
 
 
 
424
  ''')
425
  else:
426
  top_description = gr.HTML(f'''
417
  ''')
418
  else:
419
  top_description = gr.HTML(f'''
420
+ <div style="text-align: center; max-width: 650px; margin: 0 auto;">
421
+ <div>
422
+ <img class="logo" src="file/mirage.png" alt="Mirage Logo"
423
+ style="margin: auto; max-width: 7rem;">
424
+ <br />
425
+ <h1 style="font-weight: 900; font-size: 2.5rem;">
426
+ Dreambooth Training UI
427
+ </h1>
428
+ </div>
429
+ <br />
430
+ <br />
431
+ <p style="margin-bottom: 10px; font-size: 94%">
432
+ Customize Stable Diffusion v1 or v2 by giving it a few examples of a concept.
433
+ Based on the <a href="https://github.com/huggingface/diffusers">diffusers</a> implementation, additional techniques from <a href="https://github.com/TheLastBen/diffusers">TheLastBen</a> and <a href="https://github.com/ShivamShrirao/diffusers">ShivamShrirao</a>"
434
+ </p>
435
+ <br />
436
+ <p>There's only one step left before you can train your model: <a href="https://huggingface.co/spaces/{os.environ['SPACE_ID']}/settings" style="text-decoration: underline" target="_blank">attribute a <b>T4 GPU</b> to it (via the Settings tab)</a> and run the training below. Other GPUs are not compatible for now. You will be billed by the minute from when you activate the GPU until when it is turned it off.</p>
437
+ </div>
438
  ''')
439
  else:
440
  top_description = gr.HTML(f'''