add model

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+moondream filter=lfs diff=lfs merge=lfs -text

Dockerfile

@@ -0,0 +1,18 @@
+# Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
+# you will also find guides on how best to write your Dockerfile
+
+FROM python:3.9
+
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+COPY --chown=user . /app
+RUN chmod +x /app/moondream
+
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8000"]

README.md

@@ -1,11 +1,11 @@
 ---
-title: Moondream
-emoji: 🐠
-colorFrom: purple
-colorTo: indigo
+title: Vlm Recognition
+emoji: 👀
+colorFrom: yellow
+colorTo: gray
 sdk: docker
 pinned: false
-license: mit
+license: apache-2.0
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,46 @@
+import base64
+import subprocess
+from tempfile import NamedTemporaryFile
+
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+
+app = FastAPI()
+
+
+# define request body
+class RequestData(BaseModel):
+    prompt: str
+    image: str
+
+
+@app.get("/")
+def greet_json():
+    return {"Hello": "World!"}
+
+
+@app.post("/query")
+def query(data: RequestData):
+    prompt = data.prompt
+    image = data.image
+
+    try:
+        # decode base64 to image
+        image = base64.b64decode(image)
+
+        with NamedTemporaryFile(delete=True, suffix=".png") as temp_image:
+            temp_image.write(image)
+            temp_image.flush()
+
+            result = subprocess.run(
+                ["./moondream", "--prompt", prompt, "--image", temp_image.name],
+                capture_output=True,
+                text=True,
+            )
+
+            if result.returncode != 0:
+                raise Exception(result.stderr)
+
+            return {"response": result.stdout}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))

requirements.txt

@@ -0,0 +1,3 @@
+fastapi
+uvicorn[standard]
+pydantic