Jonny001 commited on
Commit
f94ea70
·
verified ·
1 Parent(s): 8228c5d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -35
app.py CHANGED
@@ -1,54 +1,29 @@
1
  import gradio as gr
2
 
3
-
4
  model_1 = gr.load("models/pimpilikipilapi1/NSFW_master")
5
  model_2 = gr.load("models/DiegoJR1973/NSFW-TrioHMH-Flux")
6
  model_3 = gr.load("models/prashanth970/flux-lora-uncensored")
7
 
8
-
9
- default_negative_prompt = (
10
- "Extra limbs, Extra fingers or toes, Disfigured face, Distorted hands, Mutated body parts, "
11
- "Missing limbs, Asymmetrical features, Blurry face, Poor anatomy, Incorrect proportions, Crooked eyes, "
12
- "Deformed hands or fingers, Double face, Unrealistic skin texture, Overly smooth skin, Poor lighting, "
13
- "Cartoonish appearance, Plastic look, Grainy, Unnatural expressions, Crossed eyes, Mutated clothing, "
14
- "Artifacts, Uncanny valley, Doll-like features, Bad symmetry, Uneven skin tones, Extra teeth, "
15
- "Unrealistic hair texture, Dark shadows on face, Overexposed face, Cluttered background, Text, watermark, "
16
- "or signature, Over-processed, Low quality, Blurry, Low resolution, Pixelated, Oversaturated, Too dark, Overexposed, Poor lighting, "
17
- "Unclear, Text or watermark, Distorted faces, Extra limbs or fingers, Disfigured, Grainy, Overly stylized, Cartoonish, "
18
- "Unrealistic anatomy, Bad proportions, Unrealistic colors, Artificial look, Background noise, Unwanted objects, Repetitive patterns, Artifacting, Abstract shapes, Out of focus"
19
- )
20
-
21
-
22
- def generate_image_model_1(prompt, negative_prompt):
23
  prompt += " 2k"
24
- try:
25
- return model_1(prompt, negative_prompt=negative_prompt)
26
- except TypeError:
27
- return model_1(prompt)
28
 
29
- def generate_image_model_2(prompt, negative_prompt):
30
  prompt += " 8k"
31
- result = model_2(prompt)
32
- if isinstance(result, tuple):
33
- return result[0]
34
- return result
35
 
36
- def generate_image_model_3(prompt, negative_prompt):
37
  prompt += " 10k"
38
- try:
39
- return model_3(prompt, negative_prompt=negative_prompt)
40
- except TypeError:
41
- return model_3(prompt)
42
 
43
  interface = gr.Interface(
44
- fn=lambda prompt, negative_prompt: (
45
- generate_image_model_1(prompt, negative_prompt),
46
- generate_image_model_2(prompt, negative_prompt),
47
- generate_image_model_3(prompt, negative_prompt)
48
  ),
49
  inputs=[
50
  gr.Textbox(label="Type your prompt here: ✍️", placeholder="Describe what you want..."),
51
- gr.Textbox(label="Negative prompt:", value=default_negative_prompt),
52
  ],
53
  outputs=[
54
  gr.Image(label="Generated Image - Model 1"),
 
1
  import gradio as gr
2
 
 
3
  model_1 = gr.load("models/pimpilikipilapi1/NSFW_master")
4
  model_2 = gr.load("models/DiegoJR1973/NSFW-TrioHMH-Flux")
5
  model_3 = gr.load("models/prashanth970/flux-lora-uncensored")
6
 
7
+ def generate_image_model_1(prompt):
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8
  prompt += " 2k"
9
+ return model_1(prompt)
 
 
 
10
 
11
+ def generate_image_model_2(prompt):
12
  prompt += " 8k"
13
+ return model_2(prompt)
 
 
 
14
 
15
+ def generate_image_model_3(prompt):
16
  prompt += " 10k"
17
+ return model_3(prompt)
 
 
 
18
 
19
  interface = gr.Interface(
20
+ fn=lambda prompt: (
21
+ generate_image_model_1(prompt),
22
+ generate_image_model_2(prompt),
23
+ generate_image_model_3(prompt)
24
  ),
25
  inputs=[
26
  gr.Textbox(label="Type your prompt here: ✍️", placeholder="Describe what you want..."),
 
27
  ],
28
  outputs=[
29
  gr.Image(label="Generated Image - Model 1"),