Update app.py

app.py

@@ -4,22 +4,28 @@
 
 import gradio as gr
 
+
 model1 = gr.load("models/pimpilikipilapi1/NSFW_master")
 model2 = gr.load("models/prashanth970/flux-lora-uncensored")
 
-
-def generate_image(text, selected_model):
+def generate_images(text, selected_model):
     if selected_model == "Model 1 (NSFW Master)":
-        result_image = model1(text)
+        model = model1
     elif selected_model == "Model 2 (Flux Lora Uncensored)":
-        result_image = model2(text)
+        model = model2
     else:
         return "Invalid model selection."
-    return result_image
-
+    
+    results = []
+    for i in range(3):
+        modified_text = f"{text} variation {i+1}"
+        result = model(modified_text)
+        results.append(result)
+    
+    return results
 
 interface = gr.Interface(
-    fn=generate_image,
+    fn=generate_images,
     inputs=[
         gr.Textbox(label="Type here your imagination:", placeholder="Type your prompt..."),
         gr.Radio(
@@ -28,9 +34,13 @@ interface = gr.Interface(
             value="Model 1 (NSFW Master)",
         ),
     ],
-    outputs=gr.Image(label="Generated Image"),
+    outputs=[
+        gr.Image(label="Generated Image 1"),
+        gr.Image(label="Generated Image 2"),
+        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 models are currently running on the CPU, which might affect performance. We appreciate your understanding.",
 )
 
 interface.launch()