GaborToth2 commited on
Commit
0269ce7
·
1 Parent(s): 37bb369

llama model switch

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -5,7 +5,8 @@ import cohere
5
 
6
  # Model & API setup
7
  COHERE_MODEL = "command-r-plus"
8
- HF_MODEL = "mistralai/Mistral-7B-Instruct-v0.3"
 
9
 
10
  # Fetch API keys from environment variables
11
  HF_API_KEY = os.getenv("HF_API_KEY")
 
5
 
6
  # Model & API setup
7
  COHERE_MODEL = "command-r-plus"
8
+ # Llama 3 3B instruct
9
+ HF_MODEL = "meta-llama/Llama-3.2-3B-Instruct"
10
 
11
  # Fetch API keys from environment variables
12
  HF_API_KEY = os.getenv("HF_API_KEY")