Updated

Dockerfile

@@ -1,5 +1,5 @@
 # ==============================================================
-# Tech Disciples AI Backend — Dockerfile
+# Tech Disciples AI Backend — Dockerfile (Authenticated Build)
 # ==============================================================
 
 # Use lightweight official Python image
@@ -13,22 +13,47 @@ ENV PYTHONDONTWRITEBYTECODE=1 \
 # Set work directory
 WORKDIR $APP_HOME
 
+# --------------------------------------------------------------
 # System dependencies
+# --------------------------------------------------------------
 RUN apt-get update && apt-get install -y \
     git \
     && rm -rf /var/lib/apt/lists/*
 
+# --------------------------------------------------------------
 # Copy requirement file first (for caching)
+# --------------------------------------------------------------
 COPY requirements.txt .
 
+# --------------------------------------------------------------
 # Install Python dependencies
+# --------------------------------------------------------------
 RUN pip install --no-cache-dir -r requirements.txt
 
+# --------------------------------------------------------------
 # Copy application files
+# --------------------------------------------------------------
 COPY . .
 
+# --------------------------------------------------------------
+# Set up Hugging Face token (must be provided at runtime)
+# --------------------------------------------------------------
+# This expects the token to be passed in during container run:
+#   docker run -e HUGGINGFACEHUB_API_TOKEN=hf_xxx -p 7860:7860 techdisciplesai
+ENV HUGGINGFACEHUB_API_TOKEN=""
+
+# --------------------------------------------------------------
+# Authenticate to Hugging Face Hub (non-interactive)
+# --------------------------------------------------------------
+RUN python -c "from huggingface_hub import login; import os; token=os.getenv('HUGGINGFACEHUB_API_TOKEN'); \
+    print('⚠️ No HF token set during build!' if not token else '✅ Hugging Face token configured.')"
+
+# --------------------------------------------------------------
 # Expose port for FastAPI
+# --------------------------------------------------------------
 EXPOSE 7860
 
+# --------------------------------------------------------------
 # Run the FastAPI app with Uvicorn
+# --------------------------------------------------------------
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -42,7 +42,7 @@ try:
 
 
     hf_pipeline = pipeline(
-        "text-generation-inference",
+        "text-generation",
         model=MODEL_NAME,
         device=DEVICE,
         max_new_tokens=1024,