srbhavya01 commited on
Commit
ee0436f
·
verified ·
1 Parent(s): 7971e0f

Update model_api.py

Browse files
Files changed (1) hide show
  1. model_api.py +3 -3
model_api.py CHANGED
@@ -6,9 +6,9 @@ def query_model(prompt):
6
  HF_TOKEN = os.getenv("HF_TOKEN")
7
 
8
  client = InferenceClient(
9
- model="microsoft/Phi-3-mini-4k-instruct",
10
- token=HF_TOKEN
11
- )
12
 
13
  response = client.chat_completion(
14
  messages=[
 
6
  HF_TOKEN = os.getenv("HF_TOKEN")
7
 
8
  client = InferenceClient(
9
+ model="HuggingFaceH4/zephyr-7b-beta",
10
+ token=HF_TOKEN
11
+ )
12
 
13
  response = client.chat_completion(
14
  messages=[