ironjr commited on
Commit
dadef83
·
verified ·
1 Parent(s): 30b6ccf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 StableMultiDiffusionPipeline
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: StableMultiDiffusionPipeline(device, sd_version='1.5', hf_key=v).cuda()
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