Update app.py

app.py

@@ -1,38 +1,32 @@
 import gradio as gr
+import random
+import os
 
-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")
 
-def generate_image_model_1(prompt):
-    prompt += " 2k"
-    return model_1(prompt)
+model1 = gr.load("models/pimpilikipilapi1/NSFW_master")
+model2 = gr.load("models/DiegoJR1973/NSFW-TrioHMH-Flux")
+model3 = gr.load("models/prashanth970/flux-lora-uncensored")
 
-def generate_image_model_2(prompt):
-    prompt += " 8k"
-    return model_2(prompt)
 
-def generate_image_model_3(prompt):
-    prompt += " 10k"
-    return model_3(prompt)
+def generate_images(text):
+    result_image1 = model1(text)
+    result_image2 = model2(text)
+    result_image3 = model3(text)
+    
+    return result_image1, result_image2, result_image3
 
 interface = gr.Interface(
-    fn=lambda prompt: (
-        generate_image_model_1(prompt),
-        generate_image_model_2(prompt),
-        generate_image_model_3(prompt)
-    ),
+    fn=generate_images,
     inputs=[
-        gr.Textbox(label="Type your prompt here: ✍️", placeholder="Describe what you want..."),
+        gr.Textbox(label="Type here your imagination:", placeholder="Type or click an example..."),
     ],
     outputs=[
-        gr.Image(label="Generated Image - Model 1"),
-        gr.Image(label="Generated Image - Model 2"),
-        gr.Image(label="Generated Image - Model 3"),
+        gr.Image(label="Model 1 Output"),
+        gr.Image(label="Model 2 Output"),
+        gr.Image(label="Model 3 Output")
     ],
-    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()