Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,6 +11,8 @@ from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
|
|
| 11 |
|
| 12 |
DESCRIPTIONx = """
|
| 13 |
|
|
|
|
|
|
|
| 14 |
## TEXT-2-IMG SDXL
|
| 15 |
"""
|
| 16 |
|
|
@@ -31,8 +33,8 @@ examples = [
|
|
| 31 |
]
|
| 32 |
|
| 33 |
MODEL_OPTIONS = {
|
| 34 |
-
"
|
| 35 |
-
"
|
| 36 |
}
|
| 37 |
|
| 38 |
MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "4096"))
|
|
@@ -155,7 +157,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
|
|
| 155 |
model_choice = gr.Dropdown(
|
| 156 |
label="Model Selection",
|
| 157 |
choices=list(MODEL_OPTIONS.keys()),
|
| 158 |
-
value="
|
| 159 |
)
|
| 160 |
|
| 161 |
with gr.Accordion("Advanced options", open=True):
|
|
|
|
| 11 |
|
| 12 |
DESCRIPTIONx = """
|
| 13 |
|
| 14 |
+
|
| 15 |
+
|
| 16 |
## TEXT-2-IMG SDXL
|
| 17 |
"""
|
| 18 |
|
|
|
|
| 33 |
]
|
| 34 |
|
| 35 |
MODEL_OPTIONS = {
|
| 36 |
+
"LightningV4": "SG161222/RealVisXL_V4.0_Lightning",
|
| 37 |
+
"RealvisionXLV4": "SG161222/RealVisXL_V4.0",
|
| 38 |
}
|
| 39 |
|
| 40 |
MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "4096"))
|
|
|
|
| 157 |
model_choice = gr.Dropdown(
|
| 158 |
label="Model Selection",
|
| 159 |
choices=list(MODEL_OPTIONS.keys()),
|
| 160 |
+
value="LightningV4"
|
| 161 |
)
|
| 162 |
|
| 163 |
with gr.Accordion("Advanced options", open=True):
|