abenkbp commited on
Commit
657820d
·
1 Parent(s): 8eeaa1e
Files changed (1) hide show
  1. data/models/llama3-1-8b.py +1 -1
data/models/llama3-1-8b.py CHANGED
@@ -10,7 +10,7 @@ app = Flask(__name__)
10
 
11
  api_key = os.getenv("ai")
12
 
13
- model_id = "meta-llama/Meta-Llama-3.1-8B-Instruct"
14
 
15
  pipeline = transformers.pipeline(
16
  "text-generation",
 
10
 
11
  api_key = os.getenv("ai")
12
 
13
+ model_id = "UCODE/agent-llama"
14
 
15
  pipeline = transformers.pipeline(
16
  "text-generation",