Upload 3 files

Dockerfile

@@ -0,0 +1,16 @@
+# 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.12
+
+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
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,19 @@
+from fastapi import FastAPI
+from transformers import AutoModel, AutoTokenizer
+
+app = FastAPI()
+
+# Load Model
+model_path = "./model"
+tokenizer = AutoTokenizer.from_pretrained(model_path)
+model = AutoModel.from_pretrained(model_path)
+
+@app.get("/")
+def home():
+    return {"message": "DeepSeek API is running!"}
+
+@app.post("/predict")
+async def predict(text: str):
+    inputs = tokenizer(text, return_tensors="pt")
+    outputs = model(**inputs)
+    return {"output": outputs.last_hidden_state.tolist()}

requirements.txt

@@ -0,0 +1,4 @@
+fastapi
+uvicorn
+torch
+transformers