Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 
 model1 = gr.load("models/Jonny001/NSFW_master")
 model2 = gr.load("models/Jonny001/Alita-v1")
-model3 = gr.load("models/lexa862/NSFWmodel")
+model3 = gr.load("models/Jonny001/WBMH-v1.1")
 model4 = gr.load("models/Keltezaa/flux_pussy_NSFW")
 model5 = gr.load("models/prashanth970/flux-lora-uncensored")
 
@@ -12,11 +12,11 @@ def generate_images(text, selected_model):
         model = model1
     elif selected_model == "Model 2 (Alita)":
         model = model2
-    elif selected_model == "Model 3 (Lexa NSFW)":
+    elif selected_model == "Model 3 (WBMH-v1.1)":
         model = model3
     elif selected_model == "Model 4 (Flux NSFW)":
         model = model4
-    elif selected_model == "Model 4 (Lora Uncensored)":
+    elif selected_model == "Model 5 (Lora Uncensored)":
         model = model5
     else:
         return "Invalid model selection."
@@ -30,11 +30,11 @@ def generate_images(text, selected_model):
     return results
 
 interface = gr.Interface(
-    fn=generate_images,#...
+    fn=generate_images,
     inputs=[
         gr.Textbox(label="Type here your imagination:", placeholder="Type your prompt..."),
         gr.Radio(
-            ["Model 1 (NSFW Master)", "Model 2 (Alita)", "Model 3 (Lexa NSFW)", "Model 4 (Flux NSFW)", "Model 5 (Lora Uncensored)"],
+            ["Model 1 (NSFW Master)", "Model 2 (Alita)", "Model 3 (WBMH-v1.1)", "Model 4 (Flux NSFW)", "Model 5 (Lora Uncensored)"],
             label="Select Model (Try All Models & Get Different Results)",
             value="Model 1 (NSFW Master)",
         ),
@@ -45,7 +45,7 @@ interface = gr.Interface(
         gr.Image(label="Generated Image 3"),
     ],
     theme="Yntec/HaleyCH_Theme_Orange",
-    description="⚠ Sorry for the inconvenience. The models are currently running on the CPU, which might affect performance. We appreciate your understanding.",
+    description="⚠ Sorry for the inconvenience. The space are currently running on the CPU, which might affect performance. We appreciate your understanding.",
     cache_examples=False,
 )