Update handler.py

handler.py

@@ -16,8 +16,9 @@ from exllamav2.generator import (
 config = ExLlamaV2Config()
 model_directory =  "/go-bruins-v2/"
 if not os.path.isdir(model_directory):
+    os.makedirs(model_directory)
     from huggingface_hub import snapshot_download
-    snapshot_download(repo_id="LoneStriker/go-bruins-v2-6.0bpw-h6-exl2-2",allow_patterns=["*.json","*.model","*.safetensors"],cache_dir=model_directory)
+    snapshot_download(repo_id="LoneStriker/go-bruins-v2-6.0bpw-h6-exl2-2",allow_patterns=["*.json","*.model","*.safetensors"],local_dir=model_directory)
 
 config.model_dir = model_directory
 config.prepare()