Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -36,7 +36,7 @@ import spaces
|
|
| 36 |
import gradio as gr
|
| 37 |
from huggingface_hub import snapshot_download
|
| 38 |
|
| 39 |
-
from model import
|
| 40 |
from util import seed_everything
|
| 41 |
from share_btn import share_js
|
| 42 |
|
|
@@ -100,7 +100,7 @@ model_dict = {
|
|
| 100 |
}
|
| 101 |
|
| 102 |
models = {
|
| 103 |
-
k:
|
| 104 |
for k, v in model_dict.items()
|
| 105 |
}
|
| 106 |
|
|
|
|
| 36 |
import gradio as gr
|
| 37 |
from huggingface_hub import snapshot_download
|
| 38 |
|
| 39 |
+
from model import SemanticDrawPipeline
|
| 40 |
from util import seed_everything
|
| 41 |
from share_btn import share_js
|
| 42 |
|
|
|
|
| 100 |
}
|
| 101 |
|
| 102 |
models = {
|
| 103 |
+
k: SemanticDrawPipeline(device, sd_version='1.5', hf_key=v).cuda()
|
| 104 |
for k, v in model_dict.items()
|
| 105 |
}
|
| 106 |
|