Update app.py

app.py

@@ -14,7 +14,8 @@ MODELS = {
     "RealVisXL V5.0 Lightning": "SG161222/RealVisXL_V5.0_Lightning",
     "Lustify Lightning": "GraydientPlatformAPI/lustify-lightning",
     "Juggernaut XL Lightning": "RunDiffusion/Juggernaut-XL-Lightning",
-    "Juggernaut-XL-V9-GE-RDPhoto2": "AiWise/Juggernaut-XL-V9-GE-RDPhoto2-Lightning_4S"
+    "Juggernaut-XL-V9-GE-RDPhoto2": "AiWise/Juggernaut-XL-V9-GE-RDPhoto2-Lightning_4S",
+    "SatPony-Lightning": "John6666/satpony-lightning-v2-sdxl"
 }
 
 config_file = hf_hub_download(
@@ -199,7 +200,8 @@ def inpaint(prompt, image, inpaint_model, paste_back):
         ).to("cuda")
     mask = Image.fromarray(image["mask"]).convert("L")
     image = Image.fromarray(image["image"])
-    result = pipe(prompt=prompt, image=image, mask_image=mask).images[0]
+    inpaint_final_prompt = f"score_9, score_8_up, score_7_up, {prompt}"
+    result = pipe(prompt=inpaint_final_prompt, image=image, mask_image=mask).images[0]
     if paste_back:
         result.paste(image, (0, 0), Image.fromarray(255 - np.array(mask)))
     return result
@@ -211,7 +213,7 @@ def outpaint(image, width, height, overlap_percentage, num_inference_steps, resi
         alignment = "Middle"
     cnet_image = background.copy()
     cnet_image.paste(0, (0, 0), mask)
-    final_prompt = f"{prompt_input} , high quality, 4k"
+    final_prompt = f"score_9, score_8_up, score_7_up, {prompt_input} , high quality, 4k"
     (
         prompt_embeds,
         negative_prompt_embeds,