Spaces:
Running
Running
Update
Browse files- app.py +2 -5
- settings.py +1 -0
app.py
CHANGED
@@ -21,13 +21,10 @@ from app_segmentation import create_demo as create_demo_segmentation
|
|
21 |
from app_shuffle import create_demo as create_demo_shuffle
|
22 |
from app_softedge import create_demo as create_demo_softedge
|
23 |
from model import Model
|
24 |
-
from settings import SHOW_DUPLICATE_BUTTON
|
25 |
|
26 |
DESCRIPTION = '# ControlNet v1.1'
|
27 |
|
28 |
-
SPACE_ID = os.getenv('SPACE_ID')
|
29 |
-
ALLOW_CHANGING_BASE_MODEL = SPACE_ID != 'hysts/ControlNet-v1-1'
|
30 |
-
|
31 |
if not torch.cuda.is_available():
|
32 |
DESCRIPTION += '\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>'
|
33 |
|
@@ -88,7 +85,7 @@ with gr.Blocks(css='style.css') as demo:
|
|
88 |
'Change base model', interactive=ALLOW_CHANGING_BASE_MODEL)
|
89 |
if not ALLOW_CHANGING_BASE_MODEL:
|
90 |
gr.Markdown(
|
91 |
-
'''The base model is not allowed to be changed in this Space so as not to slow down the demo, but it can be changed if you duplicate the Space.
|
92 |
)
|
93 |
|
94 |
check_base_model_button.click(fn=lambda: model.base_model_id,
|
|
|
21 |
from app_shuffle import create_demo as create_demo_shuffle
|
22 |
from app_softedge import create_demo as create_demo_softedge
|
23 |
from model import Model
|
24 |
+
from settings import ALLOW_CHANGING_BASE_MODEL, SHOW_DUPLICATE_BUTTON
|
25 |
|
26 |
DESCRIPTION = '# ControlNet v1.1'
|
27 |
|
|
|
|
|
|
|
28 |
if not torch.cuda.is_available():
|
29 |
DESCRIPTION += '\n<p>Running on CPU 🥶 This demo does not work on CPU.</p>'
|
30 |
|
|
|
85 |
'Change base model', interactive=ALLOW_CHANGING_BASE_MODEL)
|
86 |
if not ALLOW_CHANGING_BASE_MODEL:
|
87 |
gr.Markdown(
|
88 |
+
'''The base model is not allowed to be changed in this Space so as not to slow down the demo, but it can be changed if you duplicate the Space.'''
|
89 |
)
|
90 |
|
91 |
check_base_model_button.click(fn=lambda: model.base_model_id,
|
settings.py
CHANGED
@@ -1,5 +1,6 @@
|
|
1 |
import os
|
2 |
|
|
|
3 |
SHOW_DUPLICATE_BUTTON = os.getenv('SHOW_DUPLICATE_BUTTON') == '1'
|
4 |
|
5 |
MAX_NUM_IMAGES = int(os.getenv('MAX_NUM_IMAGES', '3'))
|
|
|
1 |
import os
|
2 |
|
3 |
+
ALLOW_CHANGING_BASE_MODEL = os.getenv('SPACE_ID') != 'hysts/ControlNet-v1-1'
|
4 |
SHOW_DUPLICATE_BUTTON = os.getenv('SHOW_DUPLICATE_BUTTON') == '1'
|
5 |
|
6 |
MAX_NUM_IMAGES = int(os.getenv('MAX_NUM_IMAGES', '3'))
|