Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -24,7 +24,7 @@ import transformers
|
|
| 24 |
from PIL import Image
|
| 25 |
from transformers import AutoModel, AutoTokenizer
|
| 26 |
|
| 27 |
-
from
|
| 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])
|