Update app.py
Browse files
app.py
CHANGED
@@ -157,7 +157,7 @@ def generate_image(
|
|
157 |
def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_available() else "cpu",
|
158 |
offload: bool = False):
|
159 |
|
160 |
-
with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange"
|
161 |
|
162 |
|
163 |
with gr.Row():
|
|
|
157 |
def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_available() else "cpu",
|
158 |
offload: bool = False):
|
159 |
|
160 |
+
with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange") as demo:
|
161 |
|
162 |
|
163 |
with gr.Row():
|