ClassificationVCAPI\nAPI en HuggingFace Space con TensorFlow Serving-like pipeline.

.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
+*.keras filter=lfs diff=lfs merge=lfs -text

Dockerfile

@@ -0,0 +1,17 @@
+FROM python:3.10
+
+# Crear usuario no-root
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+COPY --chown=user ./requirements.txt /app/requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY --chown=user . /app
+
+EXPOSE 7860
+
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

README.md

@@ -1,12 +1,11 @@
 ---
 title: ClassificationVCAPI
-emoji: 🔥
-colorFrom: yellow
-colorTo: blue
+emoji: 🚀
+colorFrom: blue
+colorTo: purple
 sdk: docker
 pinned: false
-license: cc-by-nc-4.0
-short_description: Deployment space for a computer vision model
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# ClassificationVCAPI
+API en HuggingFace Space con TensorFlow Serving-like pipeline.

app.py

@@ -0,0 +1,25 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+import tensorflow as tf
+import base64
+import numpy as np
+
+app = FastAPI(title="TF Serving Base64 API")
+
+class ImagePayload(BaseModel):
+    image_base64: str
+
+@app.post("/predict")
+def predict(payload: ImagePayload):
+    # Decodificar base64
+    img_bytes = base64.b64decode(payload.image_base64)
+    img = tf.image.decode_image(img_bytes, channels=3)
+    img = tf.image.resize(img, (224, 224))
+    img = img / 255.0
+    img = tf.expand_dims(img, 0)
+
+    # Cargar modelo local
+    model = tf.keras.models.load_model("model.keras")
+
+    pred = model.predict(img).tolist()
+    return {"prediction": pred}

model.keras

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cbd4b43027e4688614da415846066cf89e071c1a5ca8d93b09de211363c59374
+size 14732001

requirements.txt

@@ -0,0 +1,4 @@
+fastapi
+uvicorn
+tensorflow-cpu==2.17.0
+numpy

runtime.txt

@@ -0,0 +1 @@
+python-3.10