Update

app.py

@@ -2,8 +2,6 @@
 
 from __future__ import annotations
 
-import os
-
 import gradio as gr
 import torch
 
@@ -21,15 +19,14 @@ from app_segmentation import create_demo as create_demo_segmentation
 from app_shuffle import create_demo as create_demo_shuffle
 from app_softedge import create_demo as create_demo_softedge
 from model import Model
-from settings import ALLOW_CHANGING_BASE_MODEL, SHOW_DUPLICATE_BUTTON
+from settings import (ALLOW_CHANGING_BASE_MODEL, DEFAULT_MODEL_ID,
+                      SHOW_DUPLICATE_BUTTON)
 
 DESCRIPTION = '# ControlNet v1.1'
 
 if not torch.cuda.is_available():
     DESCRIPTION += '\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>'
 
-DEFAULT_MODEL_ID = os.getenv('DEFAULT_MODEL_ID',
-                             'runwayml/stable-diffusion-v1-5')
 model = Model(base_model_id=DEFAULT_MODEL_ID, task_name='Canny')
 
 with gr.Blocks(css='style.css') as demo:

settings.py

@@ -1,5 +1,8 @@
 import os
 
+DEFAULT_MODEL_ID = os.getenv('DEFAULT_MODEL_ID',
+                             'runwayml/stable-diffusion-v1-5')
+
 ALLOW_CHANGING_BASE_MODEL = os.getenv('SPACE_ID') != 'hysts/ControlNet-v1-1'
 SHOW_DUPLICATE_BUTTON = os.getenv('SHOW_DUPLICATE_BUTTON') == '1'