Spaces:
Sleeping
Sleeping
T1ckbase
commited on
Commit
·
97f7a63
1
Parent(s):
021c315
test
Browse files- Dockerfile +1 -1
- main.ts +4 -2
Dockerfile
CHANGED
|
@@ -5,7 +5,7 @@ EXPOSE 7860
|
|
| 5 |
WORKDIR /app
|
| 6 |
|
| 7 |
# Prefer not to run as root.
|
| 8 |
-
|
| 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('
|
| 110 |
await Deno.writeFile(`/tmp/${fileName}`, new Uint8Array(image), { create: true });
|
| 111 |
|
|
|
|
| 112 |
let data: any = {
|
| 113 |
-
url
|
| 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 = {
|