Update Dockerfile

Dockerfile

@@ -1,14 +1,12 @@
 FROM ghcr.io/huggingface/text-generation-inference:latest
 
-USER root  # Run container as root
+# Define a pre-existing Hugging Face model
+ENV MODEL_ID="gpt2"
 
-# Define Model
-ENV MODEL_ID="adaptiveaiventures/Llama-2-7b-chat-finetune"
-
-# Set cache directory to a writable location
+# Use a writable cache directory
 ENV HF_HOME="/tmp"
 ENV TRANSFORMERS_CACHE="/tmp"
 ENV HUGGINGFACE_HUB_CACHE="/tmp"
 
 # Run the TGI server
-CMD ["--model-id", "adaptiveaiventures/Llama-32-1B-finetuned-aaivtest5", "--port", "8080", "--num-shard", "1", "--dtype", "bfloat16", "--max-batch-prefill-tokens", "512", "--disable-custom-kernels"]
+CMD ["--model-id", "gpt2", "--port", "8080", "--num-shard", "1"]