Using BlackForestLabs/flux-schnell

app.py

@@ -96,10 +96,13 @@ def get_response(text, occasion, season, style):
 
 def get_image(llm_data):
     prompt = llm_data["query"] + "and the clothing style: " + llm_data["style"]
+    # output = replicate.run(
+    #         "stability-ai/sdxl:39ed52f2a78e934b3ba6e2a89f5b1c712de7dfea535525255b1aa35c5565e08b",
+    #         input={"prompt": prompt}
+    #         )
     output = replicate.run(
-            "stability-ai/sdxl:39ed52f2a78e934b3ba6e2a89f5b1c712de7dfea535525255b1aa35c5565e08b",
-            input={"prompt": prompt}
-            )
+        "black-forest-labs/flux-schnell",
+        input={"prompt": prompt})
     return output[0]