Update app.py

app.py

@@ -11,16 +11,16 @@ from translatepy import Translator
 MODEL = os.environ.get("MODEL")
 API_URL = "https://api-inference.huggingface.co/models/tianweiy/DMD2"
 DESCRIPTION = """
-# IMGEN🧚‍♀️DMD2
+# IMGEN🧚‍♀️
 <center>🧚‍♂️输入描述生成图片，选择喜欢的风格~<br>🧚高级选项可以设置生成数量和大小等</center>
 """
-LICENSE = "采用DMD2模型"
-
+LICENSE = "采用SDXL模型"
+MODEL_ID="stabilityai/sdxl-turbo"
 translator = Translator()
 
 
 
-def randomize_seed_fn(seed: int, randomize_seed: bool) -> float:
+def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     return seed
@@ -94,14 +94,14 @@ def generate(
     # print(client.view_api())
 
     prompt = apply_style(style, prompt)
-    strength = 0.0
+    strength = 0.7
     result = client.predict(
         None,
 		prompt,
 		seed,
         strength,
         steps,
-		api_name="/predict"
+		fn_index=1
     )
     images = result[0]
     print(images)
@@ -149,7 +149,7 @@ with gr.Blocks(css=CSS, theme="soft") as demo:
                 minimum=1,
                 maximum=10,
                 step=1,
-                value=1,
+                value=2,
             )
             seed = gr.Slider(
                 label="种子",