files new

Dockerfile

@@ -1,16 +1,17 @@
-# 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
 
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+# Create a non-root user for security (HF requirement)
 RUN useradd -m -u 1000 user
 USER user
-ENV PATH="/home/user/.local/bin:$PATH"
-
-WORKDIR /app
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
 
-COPY --chown=user ./requirements.txt requirements.txt
-RUN pip install --no-cache-dir --upgrade -r requirements.txt
+WORKDIR $HOME/app
+COPY --chown=user . $HOME/app
 
-COPY --chown=user . /app
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -1,8 +1,22 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, Query
+from wtpsplit import SaT
 
 app = FastAPI()
 
+# Load the model once during startup
+# 'sat-3l-sm' is fast and efficient for CPU Spaces
+model = SaT("sat-3l-sm")
+
 @app.get("/")
-def greet_json():
-    return {"Hello": "World!"}
-    
+def home():
+    return {"message": "SaT Segmentation API is running. Use /segment?text=your_text_here"}
+
+@app.get("/segment")
+def segment_text(text: str = Query(..., description="The text you want to split into sentences")):
+    # Perform the segmentation
+    sentences = model.split(text)
+    return {
+        "input": text,
+        "sentences": sentences,
+        "count": len(sentences)
+    }

requirements.txt

@@ -1,2 +1,4 @@
 fastapi
-uvicorn[standard]
+uvicorn
+wtpsplit
+torch