Jonny001 commited on
Commit
3e587f4
1 Parent(s): d2075dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -10
app.py CHANGED
@@ -1,18 +1,23 @@
1
- #model1 = gr.load("models/pimpilikipilapi1/NSFW_master")
2
- #model2 = gr.load("models/DiegoJR1973/NSFW-TrioHMH-Flux")
3
- #model3 = gr.load("models/prashanth970/flux-lora-uncensored")
4
-
5
  import gradio as gr
6
 
7
 
8
- model1 = gr.load("models/pimpilikipilapi1/NSFW_master")
9
- model2 = gr.load("models/prashanth970/flux-lora-uncensored")
 
 
 
10
 
11
  def generate_images(text, selected_model):
12
  if selected_model == "Model 1 (NSFW Master)":
13
  model = model1
14
- elif selected_model == "Model 2 (Flux Lora Uncensored)":
15
  model = model2
 
 
 
 
 
 
16
  else:
17
  return "Invalid model selection."
18
 
@@ -29,9 +34,9 @@ interface = gr.Interface(
29
  inputs=[
30
  gr.Textbox(label="Type here your imagination:", placeholder="Type your prompt..."),
31
  gr.Radio(
32
- ["Model 1 (NSFW Master)", "Model 2 (Flux Lora Uncensored)"],
33
- label="Select Model",
34
- value="Model 2 (Flux Lora Uncensored)",
35
  ),
36
  ],
37
  outputs=[
@@ -41,6 +46,7 @@ interface = gr.Interface(
41
  ],
42
  theme="Yntec/HaleyCH_Theme_Orange",
43
  description="⚠ Sorry for the inconvenience. The models are currently running on the CPU, which might affect performance. We appreciate your understanding.",
 
44
  )
45
 
46
  interface.launch()
 
 
 
 
 
1
  import gradio as gr
2
 
3
 
4
+ model1 = gr.load("models/Jonny001/NSFW_master")
5
+ model2 = gr.load("models/xey/sldr_flux_nsfw_v2-studio")
6
+ model3 = gr.load("models/lexa862/NSFWmodel")
7
+ model4 = gr.load("models/Keltezaa/flux_pussy_NSFW")
8
+ model5 = gr.load("models/prashanth970/flux-lora-uncensored")
9
 
10
  def generate_images(text, selected_model):
11
  if selected_model == "Model 1 (NSFW Master)":
12
  model = model1
13
+ elif selected_model == "Model 2 (Sldr Flux NSFW)":
14
  model = model2
15
+ elif selected_model == "Model 3 (Lexa NSFW)":
16
+ model = model3
17
+ elif selected_model == "Model 4 (Flux NSFW)":
18
+ model = model4
19
+ elif selected_model == "Model 4 (Lora Uncensored)":
20
+ model = model5
21
  else:
22
  return "Invalid model selection."
23
 
 
34
  inputs=[
35
  gr.Textbox(label="Type here your imagination:", placeholder="Type your prompt..."),
36
  gr.Radio(
37
+ ["Model 1 (NSFW Master)", "Model 2 (Sldr Flux NSFW)", "Model 3 (Lexa NSFW)", "Model 4 (Flux NSFW)", "Model 5 (Lora Uncensored)"],
38
+ label="Select Model (Try All Models & Get Different Results)",
39
+ value="Model 1 (NSFW Master)",
40
  ),
41
  ],
42
  outputs=[
 
46
  ],
47
  theme="Yntec/HaleyCH_Theme_Orange",
48
  description="⚠ Sorry for the inconvenience. The models are currently running on the CPU, which might affect performance. We appreciate your understanding.",
49
+ cache_examples=False,
50
  )
51
 
52
  interface.launch()