mrbeliever commited on
Commit
f3092a6
1 Parent(s): c8e34aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -7
app.py CHANGED
@@ -14,19 +14,21 @@ def generate_image(selected_model_name, text_input):
14
 
15
  # Define the models and their names
16
  models = {
17
- "Model 1": model1,
18
- "Model 2": model2,
19
- "Model 3": model3,
20
- "Model 4": model4,
21
- "Model 5": model5
22
  }
23
 
24
- # Create a dropdown to select the model
25
- model_dropdown = gr.Dropdown(choices=list(models.keys()), label="Select Model")
26
 
27
  # Create the input text box
28
  input_text = gr.Textbox(label="Input Text", placeholder="Enter text here") # Set placeholder instead of default
29
 
 
 
 
30
  # Create the output image
31
  output_image = gr.Image()
32
 
 
14
 
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
22
  }
23
 
24
+
 
25
 
26
  # Create the input text box
27
  input_text = gr.Textbox(label="Input Text", placeholder="Enter text here") # Set placeholder instead of default
28
 
29
+ # Create a dropdown to select the model
30
+ model_dropdown = gr.Dropdown(choices=list(models.keys()), label="Select Model")
31
+
32
  # Create the output image
33
  output_image = gr.Image()
34