zizytd commited on
Commit
50eea55
·
verified ·
1 Parent(s): 000b505
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,9 +9,9 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
9
 
10
  # Constants
11
  HF_TOKEN = os.environ['HUGGINGFACE_TOKEN']
12
- MODEL_NAME = "google/gemma-2-2b-it"
13
  #MODEL_NAME = "meta-llama/Llama-3.3-70B-Instruct"
14
- #MODEL_NAME = "ethicsadvisorproject/Llama-2-7b-ethical-chat-finetune"
15
  DB_DIR = 'user_data' # Directory to store individual user data
16
  os.makedirs(DB_DIR, exist_ok=True) # Ensure the directory exists
17
 
 
9
 
10
  # Constants
11
  HF_TOKEN = os.environ['HUGGINGFACE_TOKEN']
12
+ #MODEL_NAME = "google/gemma-2-2b-it"
13
  #MODEL_NAME = "meta-llama/Llama-3.3-70B-Instruct"
14
+ MODEL_NAME = "ethicsadvisorproject/Llama-2-7b-ethical-chat-finetune"
15
  DB_DIR = 'user_data' # Directory to store individual user data
16
  os.makedirs(DB_DIR, exist_ok=True) # Ensure the directory exists
17