Update app.py
Browse files
app.py
CHANGED
@@ -13,8 +13,9 @@ DESCRIPTION = """
|
|
13 |
# IMGEN🧚♀️
|
14 |
<center>🧚♂️输入描述生成图片,选择喜欢的风格~<br>🧚高级选项可以设置生成数量和大小等</center>
|
15 |
"""
|
16 |
-
LICENSE = "采用SDXL模型"
|
17 |
MODEL_ID="stabilityai/sdxl-turbo"
|
|
|
|
|
18 |
translator = Translator()
|
19 |
|
20 |
|
@@ -27,6 +28,7 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
|
|
27 |
MAX_SEED = np.iinfo(np.int32).max
|
28 |
|
29 |
client = Client(MODEL)
|
|
|
30 |
|
31 |
style_list = [
|
32 |
{
|
@@ -89,7 +91,7 @@ def generate(
|
|
89 |
prompt = str(translator.translate(prompt, 'English'))
|
90 |
print(f'prompt:{prompt}')
|
91 |
seed = int(randomize_seed_fn(seed, randomize_seed))
|
92 |
-
print(
|
93 |
|
94 |
prompt = apply_style(style, prompt)
|
95 |
strength = 0.7
|
|
|
13 |
# IMGEN🧚♀️
|
14 |
<center>🧚♂️输入描述生成图片,选择喜欢的风格~<br>🧚高级选项可以设置生成数量和大小等</center>
|
15 |
"""
|
|
|
16 |
MODEL_ID="stabilityai/sdxl-turbo"
|
17 |
+
MODEL_NAME = MODEL_ID.split("/")[-1]
|
18 |
+
LICENSE = "采用 " + MODEL_NAME +" 模型"
|
19 |
translator = Translator()
|
20 |
|
21 |
|
|
|
28 |
MAX_SEED = np.iinfo(np.int32).max
|
29 |
|
30 |
client = Client(MODEL)
|
31 |
+
client_test = Client("https://google-sdxl.hf.space/")
|
32 |
|
33 |
style_list = [
|
34 |
{
|
|
|
91 |
prompt = str(translator.translate(prompt, 'English'))
|
92 |
print(f'prompt:{prompt}')
|
93 |
seed = int(randomize_seed_fn(seed, randomize_seed))
|
94 |
+
print(client_test.view_api())
|
95 |
|
96 |
prompt = apply_style(style, prompt)
|
97 |
strength = 0.7
|