add badge when running

app.py

@@ -364,15 +364,35 @@ def check_status(top_description):
             ''')
         else:
             update_top_tag = gr.update(value=f'''
-            <div class="gr-prose" style="max-width: 80%">
-                <h2>Your model has finished training ✅</h2>
-                <p>Yay, congratulations on training your model. Scroll down to play with with it, save it (either downloading it or on the Hugging Face Hub).</p>
+            <div style="text-align: center; max-width: 650px; margin: 0 auto;">
+            <div>
+                <img class="logo" src="file/mirage.png" alt="Mirage Logo"
+                    style="margin: auto; max-width: 7rem;">
+                <br />
+                <h1 style="font-weight: 900; font-size: 2.5rem;">
+                Dreambooth Web UI
+                </h1>
+            </div>
+            <br />
+            <br />
+            <h2>Your model has finished training ✅</h2>
+            <p>Yay, congratulations on training your model. Scroll down to play with with it, save it (either downloading it or on the Hugging Face Hub).</p>
             </div>
             ''')
         show_outputs = True
     elif os.path.exists("intraining.lock"):
         update_top_tag = gr.update(value='''
-        <div class="gr-prose" style="max-width: 80%">
+        <div style="text-align: center; max-width: 650px; margin: 0 auto;">
+            <div>
+                <img class="logo" src="file/mirage.png" alt="Mirage Logo"
+                    style="margin: auto; max-width: 7rem;">
+                <br />
+                <h1 style="font-weight: 900; font-size: 2.5rem;">
+                Dreambooth Web UI
+                </h1>
+            </div>
+            <br />
+            <br />
             <h2>Don't worry, your model is still training! ⌛</h2>
             <p>You closed the tab while your model was training, but it's all good! It is still training right now. You can click the "Open logs" button above here to check the training status. Once training is done, reload this tab to interact with your model</p>
         </div>