Spaces:
Runtime error
Runtime error
Federico Galatolo
commited on
Commit
·
242d29c
1
Parent(s):
2b2edbc
Added models in dict
Browse files
app.py
CHANGED
|
@@ -22,7 +22,11 @@ category_map = {
|
|
| 22 |
}
|
| 23 |
models_map = {
|
| 24 |
"real": "Real picture",
|
| 25 |
-
"DALLE2": "DALL-E 2"
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
}
|
| 27 |
|
| 28 |
if "current" not in st.session_state:
|
|
|
|
| 22 |
}
|
| 23 |
models_map = {
|
| 24 |
"real": "Real picture",
|
| 25 |
+
"DALLE2": "DALL-E 2",
|
| 26 |
+
"Stable Diffusion": "Stable Diffusion",
|
| 27 |
+
"Latent Diffusion": "Latent Diffusion",
|
| 28 |
+
"craiyon": "Craiyon (ex dalle-mini)",
|
| 29 |
+
"GLIDE": "GLIDE"
|
| 30 |
}
|
| 31 |
|
| 32 |
if "current" not in st.session_state:
|