Jonny001 commited on
Commit
7089913
·
verified ·
1 Parent(s): 4650402

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -4,7 +4,7 @@ import gradio as gr
4
  model1 = gr.load("models/Jonny001/NSFW_master")
5
  model2 = gr.load("models/Jonny001/Alita-v1")
6
  model3 = gr.load("models/Jonny001/WBMH-v1.1")
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):
@@ -14,7 +14,7 @@ def generate_images(text, selected_model):
14
  model = model2
15
  elif selected_model == "Model 3 (WBMH-v1.1)":
16
  model = model3
17
- elif selected_model == "Model 4 (Flux NSFW)":
18
  model = model4
19
  elif selected_model == "Model 5 (Lora Uncensored)":
20
  model = model5
@@ -34,7 +34,7 @@ interface = gr.Interface(
34
  inputs=[
35
  gr.Textbox(label="Type here your imagination:", placeholder="Type your prompt..."),
36
  gr.Radio(
37
- ["Model 1 (NSFW Master)", "Model 2 (Alita)", "Model 3 (WBMH-v1.1)", "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
  ),
 
4
  model1 = gr.load("models/Jonny001/NSFW_master")
5
  model2 = gr.load("models/Jonny001/Alita-v1")
6
  model3 = gr.load("models/Jonny001/WBMH-v1.1")
7
+ model4 = gr.load("models/Jonny001/EXD-v1")
8
  model5 = gr.load("models/prashanth970/flux-lora-uncensored")
9
 
10
  def generate_images(text, selected_model):
 
14
  model = model2
15
  elif selected_model == "Model 3 (WBMH-v1.1)":
16
  model = model3
17
+ elif selected_model == "Model 4 (EXD)":
18
  model = model4
19
  elif selected_model == "Model 5 (Lora Uncensored)":
20
  model = model5
 
34
  inputs=[
35
  gr.Textbox(label="Type here your imagination:", placeholder="Type your prompt..."),
36
  gr.Radio(
37
+ ["Model 1 (NSFW Master)", "Model 2 (Alita)", "Model 3 (WBMH-v1.1)", "Model 4 (EXD)", "Model 5 (Lora Uncensored)"],
38
  label="Select Model (Try All Models & Get Different Results)",
39
  value="Model 1 (NSFW Master)",
40
  ),