change ui

app.py

@@ -31,9 +31,9 @@ maximum_concepts = 3
 
 #Pre download the files
 if(is_gpu_associated):
-    # model_v1 = snapshot_download(repo_id="multimodalart/sd-fine-tunable")
-    # model_v2 = snapshot_download(repo_id="stabilityai/stable-diffusion-2")
-    # model_v2_512 = snapshot_download(repo_id="stabilityai/stable-diffusion-2-base")
+    model_v1 = snapshot_download(repo_id="multimodalart/sd-fine-tunable")
+    model_v2 = snapshot_download(repo_id="stabilityai/stable-diffusion-2")
+    model_v2_512 = snapshot_download(repo_id="stabilityai/stable-diffusion-2-base")
     model_alt = snapshot_download(repo_id="BAAI/AltDiffusion")
     model_alt_m9 = snapshot_download(repo_id="BAAI/AltDiffusion-m9")
     safety_checker = snapshot_download(repo_id="multimodalart/sd-sc")
@@ -518,11 +518,11 @@ with gr.Blocks(css=css) as demo:
     
     #Very hacky approach to emulate dynamically created Gradio components   
     with gr.Column() as upload_your_concept:
-        with gr.Column():
+        with gr.Box():
             thing_description = gr.Markdown("You are going to train an `object`, please upload 5-10 images of the object you are planning on training on from different angles/perspectives. You must have the right to do so and you are liable for the images you use, example")
             thing_experimental = gr.Checkbox(label="Improve faces (prior preservation) - can take longer training but can improve faces", visible=False, value=False)
-            thing_image_example = gr.HTML('''<img src="file/cat-toy.png" />''')
-            things_naming = gr.Markdown("You should name your concept with a unique made up word that has low chance of the model already knowing it (e.g.: `cttoy` here). Images will be automatically cropped to 512x512.")
+            thing_image_example = gr.HTML('''<img src="https://raw.githubusercontent.com/superhero-7/img_bank/main/Naruto.png" />''')
+            things_naming = gr.Markdown("You should name your concept with a unique made up word that has low chance of the model already knowing it (e.g.: `UzNrto` here). Images will be automatically cropped to 512x512.")
             
         # with gr.Column():
         file_collection = []