Merge pull request #129 from replicate/switch-from-sdxl-to-flux

Switch from SDXL to Flux (and update deps)

Author: zeke

.gitignore

@@ -30,3 +30,5 @@ yarn-error.log*
 
 # vercel
 .vercel
+
+yarn.lock

app/api/predictions/[id]/route.js

@@ -11,8 +11,8 @@ replicate.fetch = (url, options) => {
   return fetch(url, { cache: "no-store", ...options });
 };
 
-export async function GET(request, {params}) {
-  const { id } = params;
+export async function GET(request, context) {
+  const { id } = await context.params;
   const prediction = await replicate.predictions.get(id);
 
   if (prediction?.error) {

app/api/predictions/route.js

@@ -27,7 +27,7 @@ export async function POST(request) {
   const { prompt } = await request.json();
 
   const options = {
-    version: '8beff3369e81422112d93b89ca01426147de542cd4684c244b673b105188fe5f',
+    model: 'black-forest-labs/flux-schnell',
     input: { prompt }
   }
 

app/page.js

@@ -36,7 +36,7 @@ export default function Home() {
       prediction.status !== "succeeded" &&
       prediction.status !== "failed"
     ) {
-      await sleep(1000);
+      await sleep(250);
       const response = await fetch(`/api/predictions/${prediction.id}`);
       prediction = await response.json();
       if (response.status !== 200) {
@@ -52,8 +52,8 @@ export default function Home() {
     <div className="container max-w-2xl mx-auto p-5">
       <h1 className="py-6 text-center font-bold text-2xl">
         Dream something with{" "}
-        <a href="https://replicate.com/stability-ai/sdxl?utm_source=project&utm_project=getting-started">
-          SDXL
+        <a href="https://replicate.com/black-forest-labs/flux-schnell?utm_source=project&utm_project=getting-started">
+          Flux Schnell
         </a>
       </h1>