From b317ebed866b124934406cfcbecbff5c287efd39 Mon Sep 17 00:00:00 2001
From: Atridad Lahiji <me@atri.dad>
Date: Sun, 7 Apr 2024 23:56:35 -0600
Subject: [PATCH] Ok better images JK

---
 lib/replicate.go | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/lib/replicate.go b/lib/replicate.go
index 15c8b98..83bd217 100644
--- a/lib/replicate.go
+++ b/lib/replicate.go
@@ -74,8 +74,12 @@ func ReplicateImageGeneration(prompt string, filename string) (*bytes.Buffer, er
 		"width":                  1024,
 		"height":                 1024,
 		"prompt":                 prompt,
-		"num_inference_steps":    4,
-		"negative_prompt":        "worst quality, low quality",
+		"num_inference_steps":    25,
+		"negative_prompt":        "deformed, noisy, blurry, distorted",
+		"scheduler":              "DPMSolver++",
+		"guidance_scale":         3,
+		"prompt_strength":        0.8,
+		"apply_watermark":        false,
 		"num_outputs":            1,
 		"disable_safety_checker": true,
 	}
@@ -84,7 +88,7 @@ func ReplicateImageGeneration(prompt string, filename string) (*bytes.Buffer, er
 		Events: []replicate.WebhookEventType{"start", "completed"},
 	}
 
-	prediction, predictionError := client.Run(context.Background(), "bytedance/sdxl-lightning-4step:727e49a643e999d602a896c774a0658ffefea21465756a6ce24b7ea4165eba6a", input, &webhook)
+	prediction, predictionError := client.Run(context.Background(), "playgroundai/playground-v2.5-1024px-aesthetic:a45f82a1382bed5c7aeb861dac7c7d191b0fdf74d8d57c4a0e6ed7d4d0bf7d24", input, &webhook)
 
 	if predictionError != nil {
 		return nil, predictionError