mrbeliever commited on
Commit
a2f838b
1 Parent(s): ac832f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -6,6 +6,7 @@ model2 = gr.load("models/dataautogpt3/ProteusV0.2")
6
  model3 = gr.load("models/runwayml/stable-diffusion-v1-5")
7
  model4 = gr.load("models/stablediffusionapi/juggernaut-xl-v5")
8
  model5 = gr.load("models/stabilityai/stable-diffusion-xl-base-1.0")
 
9
 
10
  # Function to switch between models and generate image
11
  def generate_image(selected_model_name, text_input):
@@ -19,6 +20,7 @@ models = {
19
  "runwayml sdxl": model3,
20
  "juggernaut-xl-v5": model4,
21
  "stable-diffusion-xl": model5
 
22
  }
23
 
24
 
 
6
  model3 = gr.load("models/runwayml/stable-diffusion-v1-5")
7
  model4 = gr.load("models/stablediffusionapi/juggernaut-xl-v5")
8
  model5 = gr.load("models/stabilityai/stable-diffusion-xl-base-1.0")
9
+ model6 = gr.load("models/Dremmar/nsfw-xl")
10
 
11
  # Function to switch between models and generate image
12
  def generate_image(selected_model_name, text_input):
 
20
  "runwayml sdxl": model3,
21
  "juggernaut-xl-v5": model4,
22
  "stable-diffusion-xl": model5
23
+ "Dreammar nsfw-xl": model6
24
  }
25
 
26