Fix open-webui/open-webui#1500

Looks like the convention update per commit 0981fae1 missed `config.py`.

Fix open-webui/open-webui#1500
Looks like the convention update per commit 0981fae1 missed `config.py`.
067c9481 · Self Denial · 916ce9bc · 067c9481
Commit 067c9481 authored Apr 13, 2024 by Self Denial
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

backend/apps/images/main.py backend/apps/images/main.py +2 -2

No files found.
--- a/backend/apps/images/main.py
+++ b/backend/apps/images/main.py
@@ -32,7 +32,7 @@ import logging
 from config import (
    SRC_LOG_LEVELS,
    CACHE_DIR,
-    IMAGE_GENERATION_ENABLED,
+    ENABLE_IMAGE_GENERATION,
    AUTOMATIC1111_BASE_URL,
    COMFYUI_BASE_URL,
 )
@@ -54,7 +54,7 @@ app.add_middleware(
 )

 app.state.ENGINE = ""
-app.state.ENABLED = IMAGE_GENERATION_ENABLED
+app.state.ENABLED = ENABLE_IMAGE_GENERATION

 app.state.OPENAI_API_KEY = ""
 app.state.MODEL = ""