Update app.py

app.py

@@ -4,7 +4,6 @@ import gradio as gr
 model_1 = gr.load("models/pimpilikipilapi1/NSFW_master")
 model_2 = gr.load("models/DiegoJR1973/NSFW-TrioHMH-Flux")
 model_3 = gr.load("models/prashanth970/flux-lora-uncensored")
-model_4 = gr.load("models/Purz/face-projection")
 
 
 default_negative_prompt = (
@@ -29,10 +28,7 @@ def generate_image_model_1(prompt, negative_prompt):
 
 def generate_image_model_2(prompt, negative_prompt):
     prompt += " 8k"
-    try:
-        return model_2(prompt, negative_prompt=negative_prompt)
-    except TypeError:
-        return model_2(prompt)
+    return model_2(prompt)  # Removed negative_prompt
 
 def generate_image_model_3(prompt, negative_prompt):
     prompt += " 10k"
@@ -41,19 +37,11 @@ def generate_image_model_3(prompt, negative_prompt):
     except TypeError:
         return model_3(prompt)
 
-def generate_image_model_4(prompt, negative_prompt):
-    try:
-        return model_4(prompt)
-    except TypeError:
-        return model_4(prompt)
-
-# Gradio interface
 interface = gr.Interface(
     fn=lambda prompt, negative_prompt: (
         generate_image_model_1(prompt, negative_prompt),
         generate_image_model_2(prompt, negative_prompt),
-        generate_image_model_3(prompt, negative_prompt),
-        generate_image_model_4(prompt, negative_prompt),
+        generate_image_model_3(prompt, negative_prompt)
     ),
     inputs=[
         gr.Textbox(label="Type your prompt here: ✍️", placeholder="Describe what you want..."),
@@ -63,10 +51,10 @@ interface = gr.Interface(
         gr.Image(label="Generated Image - Model 1"),
         gr.Image(label="Generated Image - Model 2"),
         gr.Image(label="Generated Image - Model 3"),
-        gr.Image(label="Generated Image - Model 4"),
     ],
-    title="Text to Image Generator",
-    description="⚠️ Sorry for the inconvenience. The models are currently running on the CPU, which might affect performance. We appreciate your understanding.",
+    title="Text to Image (NSFW) 🔞",
+    theme="NoCrypt/miku",
+    description="⚠️ Sorry for the inconvenience. The model is currently running on the CPU, which might affect performance. We appreciate your understanding.",
 )
 
-interface.launch()
+interface.launch()