Update app.py

app.py

@@ -4,10 +4,15 @@ import sys
 from pathlib import Path
 
 # https://huggingface.co/CompVis/stable-diffusion-v1-4
+# stabilityai/stable-diffusion-2-1
+"https://huggingface.co/stabilityai/stable-diffusion-2-1-unclip"
 
 models = [
-    {"name": "Stable Diffusion 1.4","url": "CompVis/stable-diffusion-v1-4"},
-    {"name": "Stable Diffusion 1.5","url": "runwayml/stable-diffusion-v1-5"},
+     {"name": "SD ComVis 1.2","url": "CompVis/stable-diffusion-v1-2"},
+    {"name": "SD Comvis 1.4","url": "CompVis/stable-diffusion-v1-4"},
+    {"name": "SD runawayml 1.5","url": "runwayml/stable-diffusion-v1-5"},
+    {"name": "SD Stability 2.1","url": "stabilityai/stable-diffusion-2-1"},
+    {"name": "SD Stability 2.1 Unclip","url": "stabilityai/stable-diffusion-2-1-unclip"},
     ]
 
 current_model = models[0]
@@ -43,7 +48,7 @@ with gr.Blocks (css = 'main.css')  as myface:
         input_text = gr.Textbox(label=" ",placeholder="1.PROMPT IDEA HERE ! ",lines=4)
         # Model selection dropdown
         model_name1 = gr.Dropdown(
-        label=" ",
+        label="2 Choose model here",
         choices=[m["name"] for m in models],
         type="index",
         value=current_model["name"],
@@ -52,8 +57,8 @@ with gr.Blocks (css = 'main.css')  as myface:
         
        )
     with gr.Row():
-        see_prompts = gr.Button("2. GENERATE YOUR PROMT IDEA HERE!")                                                      
-        run = gr.Button("3. GENERATE THE IMAGE HERE!", varant="primery")
+        see_prompts = gr.Button("3. GENERATE YOUR PROMT IDEA HERE!")                                                      
+        run = gr.Button("4. GENERATE THE IMAGE HERE!", varant="primery")
         
  #
     with gr.Row():