acecalisto3 commited on
Commit
9ff74fe
·
verified ·
1 Parent(s): 76e401b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -6,6 +6,13 @@ import time
6
  import logging
7
  from threading import Lock
8
 
 
 
 
 
 
 
 
9
  # Setup logging
10
  logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
11
 
 
6
  import logging
7
  from threading import Lock
8
 
9
+ CONFIG_FILE = "config.json"
10
+ MODEL_NAME = "mistralai/Mixtral-8x7B-Instruct-v0.1"
11
+ CACHE_DIR = "model_cache" # Directory for storing model cache
12
+
13
+ # Create cache directory if it doesn't exist
14
+ os.makedirs(CACHE_DIR, exist_ok=True)
15
+
16
  # Setup logging
17
  logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
18