Model download issue fix

SamurAIGPT · May 28, 2023 · 09c4399 · 09c4399
1 parent 8ab959a
commit 09c4399
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/server/privateGPT.py b/server/privateGPT.py
@@ -177,6 +177,8 @@ def download_and_save():
     filename = 'ggml-gpt4all-j-v1.3-groovy.bin'  # Specify the name for the downloaded file
     models_folder = 'models'  # Specify the name of the folder inside the Flask app root
 
+    if not os.path.exists(models_folder):
+        os.makedirs(models_folder)
     response = requests.get(url,stream=True)
     total_size = int(response.headers.get('content-length', 0))
     bytes_downloaded = 0