Spaces:

d3evil4
/

Image2Caption

Sleeping

App Files Files Community

khushalcodiste commited on Mar 8

Commit

a022cd7

1 Parent(s): 9a16713

feat: added

Browse files

Files changed (4) hide show

Dockerfile +16 -6
README.md +6 -5
docker-compose.yml +2 -2
src/server.js +39 -2

Dockerfile CHANGED Viewed

@@ -1,20 +1,30 @@
 FROM node:22-slim
-WORKDIR /app
-# sharp needs these
 RUN apt-get update && \
     apt-get install -y --no-install-recommends libvips-dev && \
     rm -rf /var/lib/apt/lists/*
-COPY package.json ./
 RUN npm install --omit=dev
-COPY src/ src/
 # Download model at build time so container starts fast
 RUN node -e "import('./src/model.js').then(m => m.loadModel()).then(() => process.exit(0))"
-EXPOSE 3000
 CMD ["node", "src/server.js"]

 FROM node:22-slim
+# sharp needs libvips
 RUN apt-get update && \
     apt-get install -y --no-install-recommends libvips-dev && \
     rm -rf /var/lib/apt/lists/*
+# HF Spaces runs as uid 1000 — create writable dirs
+RUN useradd -m -u 1000 appuser
+RUN mkdir -p /app /home/appuser/.cache && chown -R appuser:appuser /app /home/appuser/.cache
+WORKDIR /app
+COPY --chown=appuser:appuser package.json ./
 RUN npm install --omit=dev
+COPY --chown=appuser:appuser src/ src/
+# Set cache env so transformers.js writes to a writable location
+ENV TRANSFORMERS_CACHE=/home/appuser/.cache/huggingface
+ENV HF_HOME=/home/appuser/.cache/huggingface
+USER appuser
 # Download model at build time so container starts fast
 RUN node -e "import('./src/model.js').then(m => m.loadModel()).then(() => process.exit(0))"
+EXPOSE 7860
 CMD ["node", "src/server.js"]

README.md CHANGED Viewed

@@ -1,10 +1,11 @@
 ---
-title: Image2Caption
-emoji: 🏢
-colorFrom: green
-colorTo: pink
 sdk: docker
 pinned: false
 ---
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

 ---
+title: img3txt
+emoji: 📷
+colorFrom: blue
+colorTo: purple
 sdk: docker
+app_port: 7860
 pinned: false
 ---
+Image captioning API using Microsoft Florence-2 (ONNX). Open `/docs` for Swagger UI.

docker-compose.yml CHANGED Viewed

@@ -2,7 +2,7 @@ services:
   img3txt:
     build: .
     ports:
-      - "3000:3000"
     environment:
-      - PORT=3000
     restart: unless-stopped

   img3txt:
     build: .
     ports:
+      - "7860:7860"
     environment:
+      - PORT=7860
     restart: unless-stopped

src/server.js CHANGED Viewed

@@ -19,7 +19,7 @@ await app.register(swagger, {
         "Generate captions, OCR, object detection and more from images using Microsoft Florence-2 (ONNX).",
       version: "1.0.0",
     },
-    servers: [{ url: "http://localhost:3000" }],
     tags: [
       { name: "caption", description: "Image captioning endpoints" },
       { name: "health", description: "Health check" },
@@ -69,6 +69,43 @@ const errorSchema = {
 // --- Routes ---
 app.get(
   "/health",
   {
@@ -185,7 +222,7 @@ app.post(
 );
 // --- Start ---
-const PORT = process.env.PORT || 3000;
 // Pre-load model then start server
 await loadModel();

         "Generate captions, OCR, object detection and more from images using Microsoft Florence-2 (ONNX).",
       version: "1.0.0",
     },
+    servers: [{ url: "/" }],
     tags: [
       { name: "caption", description: "Image captioning endpoints" },
       { name: "health", description: "Health check" },
 // --- Routes ---
+// Landing page — HF Spaces iframe shows this
+app.get(
+  "/",
+  { schema: { hide: true } },
+  async (req, reply) => {
+    reply.type("text/html").send(`<!DOCTYPE html>
+<html lang="en"><head><meta charset="utf-8">
+<meta name="viewport" content="width=device-width,initial-scale=1">
+<title>img3txt — Florence-2 Image Captioning API</title>
+<style>
+*{margin:0;padding:0;box-sizing:border-box}
+body{font-family:system-ui,sans-serif;background:#0f172a;color:#e2e8f0;display:flex;align-items:center;justify-content:center;min-height:100vh}
+.card{background:#1e293b;border-radius:16px;padding:2.5rem;max-width:520px;width:90%;text-align:center;box-shadow:0 25px 50px rgba(0,0,0,.4)}
+h1{font-size:1.8rem;margin-bottom:.5rem}
+.sub{color:#94a3b8;margin-bottom:1.5rem}
+.btn{display:inline-block;padding:.75rem 1.5rem;background:#3b82f6;color:#fff;border-radius:8px;text-decoration:none;font-weight:600;margin:.25rem}
+.btn:hover{background:#2563eb}
+.tasks{margin-top:1.5rem;text-align:left;background:#0f172a;border-radius:8px;padding:1rem}
+.tasks code{color:#38bdf8}
+</style></head><body>
+<div class="card">
+<h1>img3txt</h1>
+<p class="sub">Image captioning, OCR &amp; object detection powered by Florence-2 (ONNX)</p>
+<a class="btn" href="/docs">Swagger UI</a>
+<a class="btn" href="/health">Health Check</a>
+<div class="tasks">
+<p><strong>POST /caption</strong> with form fields:</p>
+<ul style="margin:.5rem 0 0 1.2rem;color:#94a3b8">
+<li><code>file</code> — image (required)</li>
+<li><code>task</code> — caption, detailed_caption, more_detailed_caption, ocr, ocr_with_region, object_detection, dense_region_caption, region_proposal</li>
+<li><code>max_tokens</code> — default 100</li>
+</ul>
+</div>
+</div></body></html>`);
+  }
+);
 app.get(
   "/health",
   {
 );
 // --- Start ---
+const PORT = process.env.PORT || 7860;
 // Pre-load model then start server
 await loadModel();