Update Dockerfile

Dockerfile

@@ -1,23 +1,12 @@
-FROM arm64v8/ubuntu:jammy
+# Base image: HF text-generation-inference server
+FROM ghcr.io/huggingface/text-generation-inference:latest
 
-ENV DEBIAN_FRONTEND=noninteractive
+# Environment variables
+ENV MODEL_ID=meta-llama/Llama-2-7b-chat-hf
+ENV PORT=7860
 
-# Install Node, Python, and useful tools
-RUN apt-get update && apt-get install -y \
-    curl git sudo python3 python3-pip nodejs npm \
-    iputils-ping net-tools vim \
-    && rm -rf /var/lib/apt/lists/*
+# Expose the port for the HF Space
+EXPOSE 7860
 
-WORKDIR /app
-
-# Copy package.json first (cache layer)
-COPY package.json .
-
-# Install Node dependencies
-RUN npm install --omit=dev
-
-# Copy the app code
-COPY . .
-
-# Hugging Face will inject $PORT at runtime
-CMD ["npm", "start"]
+# Launch the HF inference server
+CMD ["text-generation-launcher", "--model-id", "meta-llama/Llama-2-7b-chat-hf", "--port", "7860"]