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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +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
- 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,7 +20,7 @@ models = {
20
  "runwayml sdxl": model3,
21
  "juggernaut-xl-v5": model4,
22
  "stable-diffusion-xl": model5
23
- "Dreammar nsfw-xl": model6
24
  }
25
 
26
 
 
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
 
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
+
24
  }
25
 
26