Update app.py

app.py

@@ -4,26 +4,34 @@ import os
 
 
 models = {
-    "Face Projection": gr.load("models/Purz/face-projection"),
     "Flux LoRA Uncensored": gr.load("models/prashanth970/flux-lora-uncensored"),
     "NSFW TrioHMH Flux": gr.load("models/DiegoJR1973/NSFW-TrioHMH-Flux"),
     "NSFW Master": gr.load("models/pimpilikipilapi1/NSFW_master")
 }
 
-def generate_image(text, num_inference_steps):
+
+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, "
+    "Deformed hands or fingers, Double face, Unrealistic skin texture, Overly smooth skin, Poor lighting, "
+    "Cartoonish appearance, Plastic look, Grainy, Unnatural expressions, Crossed eyes, Mutated clothing, "
+    "Artifacts, Uncanny valley, Doll-like features, Bad symmetry, Uneven skin tones, Extra teeth, "
+    "Unrealistic hair texture, Dark shadows on face, Overexposed face, Cluttered background, Text, watermark, "
+    "or signature, Over-processed"
+)
+
+def generate_image(text, negative_prompt):
     result_images = {}
     for model_name, model in models.items():
-        result_images[model_name] = model(text)
-
-    print(f"Inference Steps: {num_inference_steps}")
+        result_images[model_name] = model(text, negative_prompt=negative_prompt) 
     
     return [result_images[model_name] for model_name in models]
 
 interface = gr.Interface(
     fn=generate_image,
     inputs=[
-        gr.Textbox(label="Type here your imagination:", placeholder="Type or click an example..."),
-        gr.Slider(label="Number of inference steps", minimum=1, maximum=40, step=1, value=28),
+        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],
     theme="NoCrypt/miku",