Update app.py

app.py

@@ -1,11 +1,9 @@
 import gradio as gr
 
 
-models = {
-    "Gen Image 01": gr.load("models/pimpilikipilapi1/NSFW_master"),
-    "Gen Image 02": gr.load("models/DiegoJR1973/NSFW-TrioHMH-Flux"),
-    "Gen Image 03": gr.load("models/prashanth970/flux-lora-uncensored")
-}
+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")
 
 
 default_negative_prompt = (
@@ -21,29 +19,45 @@ default_negative_prompt = (
 )
 
 
-def generate_image(text, negative_prompt):
-    result_images = {}
-    for model_name, model in models.items():
-        try:
+def generate_image_model_1(prompt, negative_prompt):
+    prompt += " 1"
+    try:
+        return model_1(prompt, negative_prompt=negative_prompt)
+    except TypeError:
+        return model_1(prompt)
 
-            result_images[model_name] = model(text, negative_prompt=negative_prompt)
-        except TypeError:
-
-            result_images[model_name] = model(text)
-    
-    return [result_images[model_name] for model_name in models]
+def generate_image_model_2(prompt, negative_prompt):
+    prompt += " 2"
+    try:
+        return model_2(prompt, negative_prompt=negative_prompt)
+    except TypeError:
+        return model_2(prompt)
 
+def generate_image_model_3(prompt, negative_prompt):
+    prompt += " 3"
+    try:
+        return model_3(prompt, negative_prompt=negative_prompt)
+    except TypeError:
+        return model_3(prompt)
 
 interface = gr.Interface(
-    fn=generate_image,
+    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)
+    ),
     inputs=[
-        gr.Textbox(label="Type your prompt here:", placeholder="Describe what you want..."),
+        gr.Textbox(label="Type your prompt here: ✍️", placeholder="Describe what you want..."),
         gr.Textbox(label="Negative prompt:", value=default_negative_prompt),
     ],
-    outputs=[gr.Image(label=model_name) for model_name in models],
+    outputs=[
+        gr.Image(label="Generated Image - Model 1"),
+        gr.Image(label="Generated Image - Model 2"),
+        gr.Image(label="Generated Image - Model 3"),
+    ],
+    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.",
+    description="⚠️ Sorry for the inconvenience. The model is currently running on the CPU, which might affect performance. We appreciate your understanding.",
 )
 
-
 interface.launch()