katielink commited on
Commit
cce2bc7
·
verified ·
1 Parent(s): 75612e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -18,7 +18,8 @@ model2endpoint = {
18
  "zephyr-7b-beta": "https://api-inference.huggingface.co/models/HuggingFaceH4/zephyr-7b-beta",
19
  "mistral-7b-instruct-v0.2": "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2",
20
  "mixtral-8x7b-instruct-v0.1": "https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-Instruct-v0.1",
21
- "gemma-7b-it": "https://api-inference.huggingface.co/models/google/gemma-7b-it"
 
22
  }
23
  model_names = list(model2endpoint.keys())
24
 
 
18
  "zephyr-7b-beta": "https://api-inference.huggingface.co/models/HuggingFaceH4/zephyr-7b-beta",
19
  "mistral-7b-instruct-v0.2": "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.2",
20
  "mixtral-8x7b-instruct-v0.1": "https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-Instruct-v0.1",
21
+ "gemma-7b-it": "https://api-inference.huggingface.co/models/google/gemma-7b-it",
22
+ "llama-7b-chat": "https://api-inference.huggingface.co/models/meta-llama/Llama-2-7b-chat-hf"
23
  }
24
  model_names = list(model2endpoint.keys())
25