srustik123 commited on
Commit
811fe5d
·
verified ·
1 Parent(s): 25170c7

Update model_api.py

Browse files
Files changed (1) hide show
  1. model_api.py +1 -1
model_api.py CHANGED
@@ -6,7 +6,7 @@ def query_model(prompt):
6
  HF_TOKEN = os.getenv("HF_TOKEN")
7
 
8
  client = InferenceClient(
9
- model="meta-llama/Meta-Llama-3-8B-Instruct",
10
  token=HF_TOKEN
11
  )
12
 
 
6
  HF_TOKEN = os.getenv("HF_TOKEN")
7
 
8
  client = InferenceClient(
9
+ model="Qwen/Qwen2.5-7B-Instruct",
10
  token=HF_TOKEN
11
  )
12