T1ckbase commited on
Commit
97f7a63
·
1 Parent(s): 021c315
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. main.ts +4 -2
Dockerfile CHANGED
@@ -5,7 +5,7 @@ EXPOSE 7860
5
  WORKDIR /app
6
 
7
  # Prefer not to run as root.
8
- # USER deno
9
 
10
  RUN deno install --entrypoint main.ts
11
 
 
5
  WORKDIR /app
6
 
7
  # Prefer not to run as root.
8
+ USER deno
9
 
10
  RUN deno install --entrypoint main.ts
11
 
main.ts CHANGED
@@ -105,12 +105,14 @@ app.post('/v1/images/generations', async (c) => {
105
  const ext = response.headers.get('content-type')!.substring('image/'.length).toLowerCase();
106
  const image = await response.arrayBuffer();
107
  const fileName = `${crypto.randomUUID()}.${ext}`;
 
108
 
109
- await ensureDir('./tmp');
110
  await Deno.writeFile(`/tmp/${fileName}`, new Uint8Array(image), { create: true });
111
 
 
112
  let data: any = {
113
- url: `${headers.get('Host')}/tmp/${fileName}`,
114
  };
115
  if (params.response_format === 'b64_json') {
116
  data = {
 
105
  const ext = response.headers.get('content-type')!.substring('image/'.length).toLowerCase();
106
  const image = await response.arrayBuffer();
107
  const fileName = `${crypto.randomUUID()}.${ext}`;
108
+ const url = `${headers.get('Host')}/tmp/${fileName}`;
109
 
110
+ await ensureDir('/tmp');
111
  await Deno.writeFile(`/tmp/${fileName}`, new Uint8Array(image), { create: true });
112
 
113
+ console.log(url);
114
  let data: any = {
115
+ url,
116
  };
117
  if (params.response_format === 'b64_json') {
118
  data = {