summaryrefslogtreecommitdiffstats
path: root/krebs/5pkgs/simple/stable-generate/default.nix
blob: 23e4cfc8e3a933fb449fafab1b53aebe82043e5a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
{ pkgs, lib, ... }:

pkgs.writers.writeDashBin "stable-generate" ''
  set -efu

  export PATH=${lib.makeBinPath [
    pkgs.curl
    pkgs.jq
  ]}

  STABLE_URL=''${STABLE_URL:-http://stable-confusion.r}

  PAYLOAD=$(jq -cn --arg query "$*" '{fn_index: 51, data: [$query,"","None","None",20,"Euler a",false,false,1,1,7,-1,-1,0,0,0,false,512,512,false,0.7,0,0,"None","",false,false,false,"","Seed","","Nothing","",true,false,false,null,"",""], session_hash: "hello_this_is_dog"}')

  data=$(curl -Ssf "$STABLE_URL/run/predict/" \
    -X POST \
    --Header 'Content-Type: application/json' \
    --data "$PAYLOAD"
  )
  export data

  filename=$(jq -rn 'env.data | fromjson.data[0][0].name')

  echo "$STABLE_URL/file=$filename"
''