mrbeliever commited on
Commit
ac832f4
1 Parent(s): 34fb36b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
 
3
  # Load multiple models
4
  model1 = gr.load("models/Lykon/dreamshaper-xl-turbo")
5
- model2 = gr.load("models/hakurei/waifu-diffusion")
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")
@@ -15,7 +15,7 @@ def generate_image(selected_model_name, text_input):
15
  # Define the models and their names
16
  models = {
17
  "dreamshaper-xl-turbo": model1,
18
- "waifu-diffusion": model2,
19
  "runwayml sdxl": model3,
20
  "juggernaut-xl-v5": model4,
21
  "stable-diffusion-xl": model5
 
2
 
3
  # Load multiple models
4
  model1 = gr.load("models/Lykon/dreamshaper-xl-turbo")
5
+ 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")
 
15
  # Define the models and their names
16
  models = {
17
  "dreamshaper-xl-turbo": model1,
18
+ "ProteusV0.2": model2,
19
  "runwayml sdxl": model3,
20
  "juggernaut-xl-v5": model4,
21
  "stable-diffusion-xl": model5