Spaces:
Runtime error
Runtime error
mrbeliever
commited on
Commit
•
f3092a6
1
Parent(s):
c8e34aa
Update app.py
Browse files
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 |
-
"
|
18 |
-
"
|
19 |
-
"
|
20 |
-
"
|
21 |
-
"
|
22 |
}
|
23 |
|
24 |
-
|
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 |
|