Springboardmen commited on
Commit
16d2649
·
verified ·
1 Parent(s): fc40a2c

Update model_api.py

Browse files
Files changed (1) hide show
  1. model_api.py +1 -4
model_api.py CHANGED
@@ -5,9 +5,6 @@ def query_model(prompt):
5
  try:
6
  HF_TOKEN = os.getenv("HF_TOKEN")
7
 
8
- if not HF_TOKEN:
9
- return "⚠ HF_TOKEN not found."
10
-
11
  client = InferenceClient(
12
  model="mistralai/Mistral-7B-Instruct-v0.2",
13
  token=HF_TOKEN
@@ -18,7 +15,7 @@ def query_model(prompt):
18
  {"role": "system", "content": "You are a certified professional fitness trainer."},
19
  {"role": "user", "content": prompt}
20
  ],
21
- max_tokens=400,
22
  temperature=0.7
23
  )
24
 
 
5
  try:
6
  HF_TOKEN = os.getenv("HF_TOKEN")
7
 
 
 
 
8
  client = InferenceClient(
9
  model="mistralai/Mistral-7B-Instruct-v0.2",
10
  token=HF_TOKEN
 
15
  {"role": "system", "content": "You are a certified professional fitness trainer."},
16
  {"role": "user", "content": prompt}
17
  ],
18
+ max_tokens=600,
19
  temperature=0.7
20
  )
21