Update Dockerfile

Dockerfile

@@ -4,8 +4,6 @@ RUN pip install runpod transformers
 
 RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
 
-RUN python -c 'from transformers import pipeline; import torch; pipe = pipeline("text-generation", model="rwitz/go-bruins-v2",device=0,torch_dtype=torch.bfloat16)'
+RUN python -c 'from huggingface_hub import snapshot_download;  snapshot_download(repo_id="rwitz/go-bruins-v2")'
 
-ADD handler.py .
-
-CMD [ "python", "-u", "/handler.py" ]
+ADD handler.py .