Spaces:
Sleeping
Sleeping
testing
Browse files
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 |
-
|
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 |
|