chaojiemao commited on
Commit
161b0b1
1 Parent(s): 0f38f9c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -24,7 +24,7 @@ import transformers
24
  from PIL import Image
25
  from transformers import AutoModel, AutoTokenizer
26
 
27
- from scepter.modules.inference.ace_inference import ACEInference
28
  from scepter.modules.utils.config import Config
29
  from scepter.modules.utils.directory import get_md5
30
  from scepter.modules.utils.file_system import FS
@@ -90,7 +90,7 @@ class ChatBotUI(object):
90
  #FS.get_dir_to_local_dir("ms://AI-ModelScope/FLUX.1-dev@tokenizer/")
91
 
92
  assert len(self.model_choices) > 0
93
- if self.default_model_name == "": self.default_model_name = self.model_choices.keys()[0]
94
  self.model_name = self.default_model_name
95
  self.pipe = ACEInference()
96
  self.pipe.init_from_cfg(self.model_choices[self.default_model_name])
 
24
  from PIL import Image
25
  from transformers import AutoModel, AutoTokenizer
26
 
27
+ from ace_inference import ACEInference
28
  from scepter.modules.utils.config import Config
29
  from scepter.modules.utils.directory import get_md5
30
  from scepter.modules.utils.file_system import FS
 
90
  #FS.get_dir_to_local_dir("ms://AI-ModelScope/FLUX.1-dev@tokenizer/")
91
 
92
  assert len(self.model_choices) > 0
93
+ if self.default_model_name == "": self.default_model_name = list(self.model_choices.keys())[0]
94
  self.model_name = self.default_model_name
95
  self.pipe = ACEInference()
96
  self.pipe.init_from_cfg(self.model_choices[self.default_model_name])