Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -40,7 +40,7 @@ import gradio as gr
|
|
40 |
from huggingface_hub import snapshot_download
|
41 |
import spaces
|
42 |
|
43 |
-
from model import
|
44 |
from util import seed_everything
|
45 |
from prompt_util import preprocess_prompts, _quality_dict, _style_dict
|
46 |
from share_btn import share_js
|
@@ -108,7 +108,7 @@ else:
|
|
108 |
model_dict = {os.path.splitext(os.path.basename(opt.model))[0]: opt.model}
|
109 |
|
110 |
models = {
|
111 |
-
k:
|
112 |
for k, v in model_dict.items()
|
113 |
}
|
114 |
|
|
|
40 |
from huggingface_hub import snapshot_download
|
41 |
import spaces
|
42 |
|
43 |
+
from model import SemanticDrawSDXLPipeline
|
44 |
from util import seed_everything
|
45 |
from prompt_util import preprocess_prompts, _quality_dict, _style_dict
|
46 |
from share_btn import share_js
|
|
|
108 |
model_dict = {os.path.splitext(os.path.basename(opt.model))[0]: opt.model}
|
109 |
|
110 |
models = {
|
111 |
+
k: SemanticDrawSDXLPipeline(device, hf_key=v, has_i2t=False).cuda()
|
112 |
for k, v in model_dict.items()
|
113 |
}
|
114 |
|