stockholm/krebs/5pkgs/simple/stable-interrogate/default.nix

31 lines
641 B
Nix
Raw Normal View History

2022-12-13 14:39:44 +00:00
{ pkgs, lib, ... }:
pkgs.writers.writeBashBin "stable-interrogate" ''
set -xefu
set -o pipefail
export PATH=${lib.makeBinPath [
pkgs.coreutils
pkgs.curl
pkgs.jq
]}
STABLE_URL=''${STABLE_URL:-http://stable-confusion.r}
(if test -e "$1"; then
cat "$1"
elif [[ "$1" =~ ^https?: ]]; then
curl -fSs "$1"
else
echo 'input not recognized' >&2
exit 1
fi) | base64 |
jq -Rsrc '{
image: .,
model: "deepdanbooru", # clip is broken right now :(
}' |
curl -Ssf "$STABLE_URL/sdapi/v1/interrogate" \
-X POST -H 'Content-Type: application/json' -d @- |
jq -r '.caption'
''