Update app_v3.py

app_v3.py

@@ -11,9 +11,9 @@ cwd = os.getcwd()
 quantized_model_dir = cwd + "/Jackson2-4bit-128g-GPTQ"
 
 # Check if the model directory is empty (i.e., model not downloaded yet)
-if not os.listdir(quantized_model_dir):
+if not os.path.exists(quantized_model_dir) or not os.listdir(quantized_model_dir):
     # Create the cache directory if it doesn't exist
-    os.makedirs(quantized_model_dir)
+    os.makedirs(quantized_model_dir, exist_ok=True)
     snapshot_download(repo_id=pretrained_model_dir, local_dir=quantized_model_dir, local_dir_use_symlinks=True)
 
 st.write(f'{os.listdir(quantized_model_dir)}')