Spaces:
Runtime error
Runtime error
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 |
|