Update app.py

app.py

@@ -5,7 +5,6 @@ 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 = (
     "Extra limbs, Extra fingers or toes, Disfigured face, Distorted hands, Mutated body parts, "
     "Missing limbs, Asymmetrical features, Blurry face, Poor anatomy, Incorrect proportions, Crooked eyes, "
@@ -19,32 +18,24 @@ default_negative_prompt = (
 )
 
 
-def generate_image_model_1(prompt, negative_prompt):
-    prompt += " 2k"
+def generate_image(model, prompt, negative_prompt):
+    """Generate image using the given model."""
+    prompt += " high-resolution"
     try:
-        return model_1(prompt, negative_prompt=negative_prompt)
+        output = model(prompt, negative_prompt=negative_prompt)
     except TypeError:
-        return model_1(prompt)
+        output = model(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)
+    if isinstance(output, tuple) and len(output) > 0:
+        output = output[0]
+    return output
 
-def generate_image_model_3(prompt, negative_prompt):
-    prompt += " 10k"
-    try:
-        return model_3(prompt, negative_prompt=negative_prompt)
-    except TypeError:
-        return model_3(prompt)
 
 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_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..."),
@@ -56,8 +47,7 @@ interface = gr.Interface(
         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.",
 )
 
-interface.launch() 
+interface.launch()