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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,7 +5,7 @@ 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):
11
  if selected_model == "Model 1 (NSFW Master)":
@@ -16,7 +16,7 @@ def generate_images(text, selected_model):
16
  model = model3
17
  elif selected_model == "Model 4 (EXD)":
18
  model = model4
19
- elif selected_model == "Model 5 (Lora Uncensored)":
20
  model = model5
21
  else:
22
  return "Invalid model selection."
@@ -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 (EXD)", "Model 5 (Lora Uncensored)"],
38
  label="Select Model (Try All Models & Get Different Results)",
39
  value="Model 1 (NSFW Master)",
40
  ),
 
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/Jonny001/R.I.O.W")
9
 
10
  def generate_images(text, selected_model):
11
  if selected_model == "Model 1 (NSFW Master)":
 
16
  model = model3
17
  elif selected_model == "Model 4 (EXD)":
18
  model = model4
19
+ elif selected_model == "Model 5 (R.I.O.W)":
20
  model = model5
21
  else:
22
  return "Invalid model selection."
 
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 (R.I.O.W)"],
38
  label="Select Model (Try All Models & Get Different Results)",
39
  value="Model 1 (NSFW Master)",
40
  ),