lodist commited on
Commit
ca95caf
·
verified ·
1 Parent(s): 5d6ab25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -7
app.py CHANGED
@@ -10,13 +10,12 @@ HF_API_TOKEN = os.getenv("HF_TOKEN")
10
 
11
  def get_hf_model():
12
  return InferenceClientModel(
13
- model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
14
- # provider="auto", # You can omit this, 'auto' is the default
15
- token=HF_API_TOKEN, # Use 'token' instead of 'api_key'
16
- temperature=0.2,
17
- max_tokens=8192,
18
- timeout=120,
19
- )
20
 
21
  # Make sure to update this line too
22
  MODEL = get_hf_model()
 
10
 
11
  def get_hf_model():
12
  return InferenceClientModel(
13
+ model_id="mistralai/Mistral-7B-Instruct-v0.3",
14
+ token=HF_API_TOKEN,
15
+ temperature=0.2,
16
+ max_tokens=4096, # Smaller models have smaller context windows
17
+ timeout=120,
18
+ )
 
19
 
20
  # Make sure to update this line too
21
  MODEL = get_hf_model()