Upload folder using huggingface_hub

Dockerfile

@@ -3,6 +3,7 @@ FROM python:3.10-slim
 WORKDIR /app
 
 COPY requirements.txt .
+RUN pip install --no-cache-dir torch==2.1.0 --index-url https://download.pytorch.org/whl/cpu
 RUN pip install --no-cache-dir -r requirements.txt
 
 COPY handler.py .

handler.py

@@ -187,10 +187,24 @@ class TextFeatureExtractorEndpoint:
 # FastAPI handler for deployment
 # ──────────────────────────────────────────────────────────────────────── #
 
-from fastapi import FastAPI
+from fastapi import FastAPI, Request
 from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import JSONResponse
 from pydantic import BaseModel
-from typing import Optional
+from typing import Optional, List, Dict
+import traceback
+
+# ──────────────────────────────────────────────────────────────────────── #
+# Constants & Defaults
+# ──────────────────────────────────────────────────────────────────────── #
+
+DEFAULT_TEXT_FEATURES = {
+    "t0_explicit_free": 0.0, "t1_explicit_busy": 0.0,
+    "t2_avg_resp_len": 0.0, "t3_short_ratio": 0.0,
+    "t4_cognitive_load": 0.0, "t5_time_pressure": 0.0,
+    "t6_deflection": 0.0, "t7_sentiment": 0.0,
+    "t8_coherence": 0.5, "t9_latency": 0.0,
+}
 
 app = FastAPI(title="Text Feature Extraction API", version="1.0.0")
 app.add_middleware(
@@ -199,6 +213,16 @@ app.add_middleware(
     allow_methods=["*"], allow_headers=["*"],
 )
 
+
+@app.exception_handler(Exception)
+async def global_exception_handler(request: Request, exc: Exception):
+    print(f"[GLOBAL ERROR] {request.url}: {exc}")
+    traceback.print_exc()
+    return JSONResponse(
+        status_code=200,
+        content={**DEFAULT_TEXT_FEATURES, "_error": str(exc), "_handler": "global"},
+    )
+
 extractor = TextFeatureExtractorEndpoint()
 
 

requirements.txt

@@ -1,7 +1,7 @@
 # NLP
 transformers==4.35.0
 sentence-transformers==2.2.2
-torch==2.1.0
+
 numpy==1.24.3
 scikit-learn==1.3.2