Sreerama commited on
Commit
b672101
1 Parent(s): e435f17

update description for gpu associated

Browse files
Files changed (1) hide show
  1. app.py +18 -4
app.py CHANGED
@@ -410,10 +410,24 @@ with gr.Blocks(css=css) as demo:
410
  elif(is_spaces):
411
  if(is_gpu_associated):
412
  top_description = gr.HTML(f'''
413
- <div class="gr-prose" style="max-width: 80%">
414
- <h2>You have successfully associated a GPU to the Dreambooth Training Space 🎉</h2>
415
- <p>Certify that you got a T4. You can now train your model! You will be billed by the minute from when you activated the GPU until when it is turned it off.</p>
416
- </div>
 
 
 
 
 
 
 
 
 
 
 
 
 
 
417
  ''')
418
  else:
419
  top_description = gr.HTML(f'''
410
  elif(is_spaces):
411
  if(is_gpu_associated):
412
  top_description = gr.HTML(f'''
413
+ <div style="text-align: center; max-width: 650px; margin: 0 auto;">
414
+ <div>
415
+ <img class="logo" src="file/mirage.png" alt="Mirage Logo"
416
+ style="margin: auto; max-width: 7rem;">
417
+ <br />
418
+ <h1 style="font-weight: 900; font-size: 2.5rem;">
419
+ Dreambooth Web UI
420
+ </h1>
421
+ </div>
422
+ <br />
423
+ <br />
424
+ <p style="margin-bottom: 10px; font-size: 94%">
425
+ Customize Stable Diffusion v1 or v2 by giving it a few examples of a concept.
426
+ 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>"
427
+ </p>
428
+ <br />
429
+ <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>
430
+ </div>
431
  ''')
432
  else:
433
  top_description = gr.HTML(f'''